Notebook opened in colab cannot run successfully

[matchyc]

I think the quickstart and tutorial notebook will get stuck at the very beginning.

[PredictiveManish]

Can you mention more details either any short video or screenshots!

[AryanVBW]

This is likely a dependency version mismatch issue that has affected multiple users. Here are potential workarounds:

Option 1: Update numpy first

%pip install --upgrade numpy

Then restart the runtime and re-run.

Option 2: Install with legacy resolver

pip install jaxlib==0.4.20
pip install git+https://github.com/deepmind/acme.git#egg=dm-acme[tf,envs] --use-deprecated=legacy-resolver

Option 3: Specify compatible versions
The Acme package has known issues with JAX version compatibility (see #321, #317). You may need to manually pin versions:

jax>=0.4.13
jaxlib>=0.4.13  
chex==0.1.7

Note: Acme's dependencies have not been updated to support the latest JAX versions, which causes these Colab errors. Until the package dependencies are updated, installing from git with pinned versions is the most reliable workaround.

Could you share a screenshot or more specific error details to help narrow down the exact issue?

[natinew77-creator]

Hi there! I encountered this same issue with the strict version pinning in setup.py conflicting with Colab's pre-installed environment.

I've submitted a PR (#356) that relaxes these constraints (changing == to >= for jax, tensorflow, etc.) which resolves the installation errors.

In the meantime, a workaround is to clone the repo and manually edit setup.py to loosen the version requirements before installing.