Update to Jumpy 1.0 (#345)

pseudo-rnd-thoughts · web-flow · commit 24a55188e5a8 · 2023-03-17T17:41:45.000Z
diff --git a/gymnasium/experimental/wrappers/lambda_action.py b/gymnasium/experimental/wrappers/lambda_action.py
@@ -10,7 +10,7 @@
 
 
 try:
-    import jumpy as jp
+    import jumpy.numpy as jp
 except ImportError as e:
     raise ImportError("Jumpy is not installed, run `pip install jax-jumpy`") from e
 import numpy as np
diff --git a/gymnasium/experimental/wrappers/lambda_observations.py b/gymnasium/experimental/wrappers/lambda_observations.py
@@ -18,7 +18,7 @@
 
 
 try:
-    import jumpy as jp
+    import jumpy.numpy as jp
 except ImportError as e:
     raise ImportError("Jumpy is not installed, run `pip install jax-jumpy`") from e
 import numpy as np
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ classifiers = [
 ]
 dependencies = [
     "numpy >=1.21.0",
-    "jax-jumpy >=0.2.0",
+    "jax-jumpy>=1.0.0",
     "cloudpickle >=1.2.0",
     "importlib-metadata >=4.8.0; python_version < '3.10'",
     "typing-extensions >=4.3.0",

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ classifiers = [`
`26`	`26`	`]`
`27`	`27`	`dependencies = [`
`28`	`28`	`"numpy >=1.21.0",`
`29`		`- "jax-jumpy >=0.2.0",`
	`29`	`+ "jax-jumpy>=1.0.0",`
`30`	`30`	`"cloudpickle >=1.2.0",`
`31`	`31`	`"importlib-metadata >=4.8.0; python_version < '3.10'",`
`32`	`32`	`"typing-extensions >=4.3.0",`