From 72b4669563df2fd879aa54e12cbb8aa2e37fb104 Mon Sep 17 00:00:00 2001
From: henk717 <henk@henk.tech>
Date: Thu, 30 Mar 2023 23:41:35 +0200
Subject: [PATCH] Fix the chex dependency

---
 requirements_mtj.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index e66e1ba7..9447541f 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -5,6 +5,7 @@ requests
 dm-haiku == 0.0.5
 jax == 0.2.21
 jaxlib >= 0.1.69, <= 0.3.7
+chex == 0.1.5
 transformers == 4.24.0
 huggingface_hub==0.12.1
 progressbar2