Pinning dm-haiku to 0.0.10

The recent 0.0.11 release of dm-haiku seems to break some examples (including the Acme quickstart Colab notebook). Pinning to the previous version solves the issue for now
google-deepmind · Nov 17, 2023 · 6d7d7a4 · 6d7d7a4
1 parent 1177501
commit 6d7d7a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -57,7 +57,7 @@
     'jax==0.4.3',
     'jaxlib==0.4.3',
     'chex',
-    'dm-haiku',
+    'dm-haiku==0.0.10',
     'flax',
     'optax',
     'rlax',