Re-deprecate a number of symbols from jax.core

jax-ml · Dec 16, 2024 · b105a08 · b105a08
1 parent 2b06f93
commit b105a08
Showing 1 changed file with 22 additions and 22 deletions.
diff --git a/jax/core.py b/jax/core.py
@@ -88,28 +88,28 @@
 
 from jax._src import core as _src_core
 _deprecations = {
-    # TODO(jakevdp): re-deprecate these after migrating some downstream uses.
-    # "ClosedJaxpr": ("jax.core.ClosedJaxpr is deprecated. Use jax.extend.core.ClosedJaxpr instead, "
-    #                 "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #                 _src_core.ClosedJaxpr),
-    # "Jaxpr": ("jax.core.Jaxpr is deprecated. Use jax.extend.core.Jaxpr instead, "
-    #           "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #           _src_core.Jaxpr),
-    # "JaxprEqn": ("jax.core.JaxprEqn is deprecated. Use jax.extend.core.JaxprEqn instead, "
-    #              "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #              _src_core.JaxprEqn),
-    # "Literal": ("jax.core.Literal is deprecated. Use jax.extend.core.Literal instead, "
-    #             "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #             _src_core.Literal),
-    # "Primitive": ("jax.core.Primitive is deprecated. Use jax.extend.core.Primitive instead, "
-    #               "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #               _src_core.Primitive),
-    # "Token": ("jax.core.Token is deprecated. Use jax.extend.core.Token instead, "
-    #           "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #           _src_core.Token),
-    # "Var": ("jax.core.Var is deprecated. Use jax.extend.core.Var instead, "
-    #         "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
-    #         _src_core.Var),
+    # Added 2024-12-16
+    "ClosedJaxpr": ("jax.core.ClosedJaxpr is deprecated. Use jax.extend.core.ClosedJaxpr instead, "
+                    "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+                    _src_core.ClosedJaxpr),
+    "Jaxpr": ("jax.core.Jaxpr is deprecated. Use jax.extend.core.Jaxpr instead, "
+              "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+              _src_core.Jaxpr),
+    "JaxprEqn": ("jax.core.JaxprEqn is deprecated. Use jax.extend.core.JaxprEqn instead, "
+                 "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+                 _src_core.JaxprEqn),
+    "Literal": ("jax.core.Literal is deprecated. Use jax.extend.core.Literal instead, "
+                "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+                _src_core.Literal),
+    "Primitive": ("jax.core.Primitive is deprecated. Use jax.extend.core.Primitive instead, "
+                  "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+                  _src_core.Primitive),
+    "Token": ("jax.core.Token is deprecated. Use jax.extend.core.Token instead, "
+              "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+              _src_core.Token),
+    "Var": ("jax.core.Var is deprecated. Use jax.extend.core.Var instead, "
+            "and see https://jax.readthedocs.io/en/latest/jax.extend.html for details.",
+            _src_core.Var),
     # Added 2024-12-11
     "axis_frame": ("jax.core.axis_frame is deprecated.", _src_core.axis_frame),
     "AxisName": ("jax.core.AxisName is deprecated.", _src_core.AxisName),