From 3019726e6d1ce203ac7c0229360df2e8ae3a8296 Mon Sep 17 00:00:00 2001
From: John Agapiou <jagapiou@google.com>
Date: Mon, 12 Feb 2024 06:33:02 -0800
Subject: [PATCH] remove restriction on chex

PiperOrigin-RevId: 606233485
Change-Id: I8f8f76061171fff41f5096fb55367d1a225ce02b
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 199ef5ff..160e7ddb 100644
--- a/setup.py
+++ b/setup.py
@@ -103,7 +103,7 @@ def build_assets(self):
     python_requires='>=3.11',
     install_requires=[
         'absl-py',
-        'chex<0.1.81',  # Incompatible with tensorflow 2.13 (due to numpy req).
+        'chex',
         'dm-env',
         'dmlab2d',
         'dm-tree',