accelerate deprecation of jax.curry

PiperOrigin-RevId: 520958381
This commit is contained in:
Jake VanderPlas 2023-03-31 10:36:58 -07:00 committed by jax authors
parent ffb8352848
commit b37c741c6f

View File

@ -9715,8 +9715,6 @@ class DeprecationsTest(jtu.JaxTestCase):
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.curry, jax._src.util.curry)
with self.assertWarns(DeprecationWarning):
self.assertIs(jax.flatten_fun_nokwargs, jax.api_util.flatten_fun_nokwargs)