From 8b78b8d9f9032d1aad0ad0a337742e8529560319 Mon Sep 17 00:00:00 2001
From: Shaobo Hou <shaobohou@google.com>
Date: Thu, 21 Dec 2023 11:49:09 -0800
Subject: [PATCH] Update chex and haiku version

PiperOrigin-RevId: 592916091
---
 requirements_tests.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements_tests.txt b/requirements_tests.txt
index 4e861ce..f964173 100644
--- a/requirements_tests.txt
+++ b/requirements_tests.txt
@@ -1,5 +1,5 @@
-chex==0.1.8
-dm-haiku==0.0.10
+chex==0.1.85
+dm-haiku==0.0.11
 dm-sonnet==2.0.1
 tensorflow==2.11.0; sys_platform != 'darwin' or platform_machine != 'arm64'
 tensorflow-macos==2.11.0; sys_platform == 'darwin' and platform_machine == 'arm64'