mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove deprecation warning test in preparation for removing deprecated APIs

Browse Source 
PiperOrigin-RevId: 547229078
This commit is contained in:
Jake VanderPlas 2023-07-11 10:26:53 -07:00 committed by jax authors
parent b581ad1f33
commit a29d4bcd33
1 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/api_test.py
View File

@ -10073,19 +10073,6 @@ class GarbageCollectionTest(jtu.JaxTestCase):
assert x_np_weakref() is None
class DeprecationsTest(jtu.JaxTestCase):
def test_jax_deprecations(self):
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.ad, jax.interpreters.ad)
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.partial_eval, jax.interpreters.partial_eval)
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.pxla, jax.interpreters.pxla)
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.xla, jax.interpreters.xla)
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.flatten_fun_nokwargs, jax.api_util.flatten_fun_nokwargs)
class OverrideLoweringTest(jtu.JaxTestCase):