From 50a96485a97a418bf70b31e1abb44fd7027b60cb Mon Sep 17 00:00:00 2001
From: henk717 <henk717@msn.com>
Date: Mon, 14 Feb 2022 18:05:50 +0100
Subject: [PATCH] Fix dm-haiku

They did a change that breaks compatibility with our other dependencies, forcing version 0.0.5 to fix this.
---
 requirements_mtj.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index d9b403eb..ad205326 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -2,7 +2,7 @@ numpy
 tqdm
 requests
 optax >= 0.0.5, <= 0.0.9
-dm-haiku
+dm-haiku == 0.0.5
 ray[default]
 jax == 0.2.21
 transformers
@@ -15,4 +15,4 @@ flask-ngrok
 eventlet
 lupa==1.10
 markdown
-bleach
\ No newline at end of file
+bleach