From 033d247a45a14bbdc862fb4dd6e9a73cf756114c Mon Sep 17 00:00:00 2001 From: natinew77-creator Date: Sun, 7 Dec 2025 11:44:04 -0500 Subject: [PATCH] Fixes #347: Pin JAX/Chex dependencies in Colab notebooks The quickstart and tutorial notebooks fail to run in Google Colab due to dependency version mismatches with JAX and Chex libraries. This fix pins compatible versions before installing dm-acme: - jax>=0.4.13 - jaxlib>=0.4.13 - chex==0.1.7 Closes #347 --- examples/quickstart.ipynb | 2 ++ examples/tutorial.ipynb | 2 ++ 2 files changed, 4 insertions(+) diff --git a/examples/quickstart.ipynb b/examples/quickstart.ipynb index f245bb414f..2b53d473e0 100644 --- a/examples/quickstart.ipynb +++ b/examples/quickstart.ipynb @@ -41,6 +41,8 @@ }, "outputs": [], "source": [ + "# Fix for issue #347: Pin compatible dependency versions\n", + "%pip install jax>=0.4.13 jaxlib>=0.4.13 chex==0.1.7\n", "%pip install git+https://github.com/deepmind/acme.git#egg=dm-acme[jax,tf,envs]" ] }, diff --git a/examples/tutorial.ipynb b/examples/tutorial.ipynb index f57b3c4942..45a30fe2e2 100644 --- a/examples/tutorial.ipynb +++ b/examples/tutorial.ipynb @@ -39,6 +39,8 @@ }, "outputs": [], "source": [ + "# Fix for issue #347: Pin compatible dependency versions\n", + "%pip install jax>=0.4.13 jaxlib>=0.4.13 chex==0.1.7\n", "%pip install git+https://github.com/deepmind/acme.git#egg=dm-acme[jax,tf,envs]" ] },