From 6d7d7a4a2339efd01b7384f7de0dcece5faa5581 Mon Sep 17 00:00:00 2001 From: jonas-eschmann Date: Fri, 17 Nov 2023 14:53:08 -0500 Subject: [PATCH] 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 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index bd46a62cb2..6a49fa72ed 100755 --- 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',