Update python_extra_deps.sh

google-deepmind · Nov 3, 2023 · e23fd12 · e23fd12
1 parent a882d52
commit e23fd12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/open_spiel/scripts/python_extra_deps.sh b/open_spiel/scripts/python_extra_deps.sh
@@ -24,7 +24,7 @@
 #
 # To enable specific tests, please use the environment variables found in
 # scripts/global_variables.sh
-export OPEN_SPIEL_PYTHON_JAX_DEPS="jax==0.4.10 jaxlib==0.4.10 dm-haiku==0.0.10 optax==0.1.7 chex==0.1.7 rlax==0.1.5 distrax==0.1.3"
+export OPEN_SPIEL_PYTHON_JAX_DEPS="jax==0.4.2 jaxlib==0.4.2 dm-haiku==0.0.10 optax==0.1.7 chex==0.1.7 rlax==0.1.5 distrax==0.1.3"
 export OPEN_SPIEL_PYTHON_PYTORCH_DEPS="torch==1.13.1"
-export OPEN_SPIEL_PYTHON_TENSORFLOW_DEPS="numpy==1.24.3 tensorflow==2.13.1 tensorflow-probability==0.19.0 tensorflow_datasets==4.9.2 keras==2.13.1"
+export OPEN_SPIEL_PYTHON_TENSORFLOW_DEPS="numpy==1.23.5 tensorflow==2.13.1 tensorflow-probability==0.19.0 tensorflow_datasets==4.9.2 keras==2.13.1"
 export OPEN_SPIEL_PYTHON_MISC_DEPS="IPython==5.8.0 networkx==2.4 matplotlib==3.5.2 mock==4.0.2 nashpy==0.0.19 scipy==1.10.1 testresources==2.0.1 cvxopt==1.3.1 cvxpy==1.2.0 ecos==2.0.10 osqp==0.6.2.post5 clu==0.0.6 flax==0.5.3"