Removed the no longer necessary jax-aval-named-shape deprecation

superbobry · Google-ML-Automation · commit fa3b7b9e405b · 2025-09-23T08:02:51.000-07:00
PiperOrigin-RevId: 810441482
diff --git a/jax/_src/array.py b/jax/_src/array.py
@@ -27,7 +27,6 @@
 from jax._src import basearray
 from jax._src import config
 from jax._src import core
-from jax._src import deprecations
 from jax._src import dispatch
 from jax._src import dtypes
 from jax._src import errors
@@ -124,16 +123,6 @@ def _reconstruct_array(fun, args, arr_state, aval_state):
   np_value = fun(*args)
   np_value.__setstate__(arr_state)
   jnp_value = api.device_put(np_value)
-  # TODO(slebedev): Remove this branch after December 10th 2024.
-  if "named_shape" in aval_state:
-    deprecations.warn(
-        "jax-aval-named-shape",
-        "Pickled array contains an aval with a named_shape attribute. This is"
-        " deprecated and the code path supporting such avals will be removed."
-        " Please re-pickle the array.",
-        stacklevel=2,
-    )
-    del aval_state["named_shape"]
   jnp_value.aval = jnp_value.aval.update(**aval_state)
   return jnp_value
 
diff --git a/jax/_src/deprecations.py b/jax/_src/deprecations.py
@@ -124,7 +124,6 @@ def warn(deprecation_id: str, message: str, stacklevel: int) -> None:
 # Register a number of deprecations: we do this here to ensure they're
 # always registered by the time `accelerate` and `is_acelerated` are called.
 register('default-dtype-bits-config')
-register('jax-aval-named-shape')
 register('jax-lax-dot-positional-args')
 register('jax-nn-one-hot-float-input')
 register("jax-numpy-astype-complex-to-real")
