fixed dep

peng-lab · Mar 6, 2024 · 54f4dce · 54f4dce
1 parent f41bd95
commit 54f4dce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -23,8 +23,8 @@ project_urls =
 packages = find:
 install_requires =
     hyperactive>=4.4.0
-    jax>=0.3.10 #,<=0.3.23
-    jaxlib>=0.3.10 #,<=0.3.23 # to import jaxlib.xla_extension.XlaRuntimeError
+    jax>=0.3.10,<=0.3.23,<=0.4.23  #
+    jaxlib>=0.3.10,<=0.3.23 # to import jaxlib.xla_extension.XlaRuntimeError,<=0.4.23  #
     numpy
     pooch
     pydantic>=1.9.1,<2.0.0
@@ -44,7 +44,7 @@ console_scripts =
 [options.extras_require]
 bench =
     jax
-    jaxlib>=0.3.10 # to import jaxlib.xla_extension.XlaRuntimeError
+    jaxlib>=0.3.10,<=0.4.23  # to import jaxlib.xla_extension.XlaRuntimeError
     scipy
 dask =
     dask
@@ -61,7 +61,7 @@ dev =
     flake8-markdown
     flake8-pytest
     isort
-    jax
+    jax>=0.3.10,<=0.3.23,<=0.4.23  #
     jaxlib>=0.3.10,<=0.4.23 # to import jaxlib.xla_extension.XlaRuntimeError
     mypy
     nox