diff --git a/tests/api_test.py b/tests/api_test.py index 18acaea36..e5d06d1a9 100644 --- a/tests/api_test.py +++ b/tests/api_test.py @@ -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):