remove restriction on chex

PiperOrigin-RevId: 606233485 Change-Id: I8f8f76061171fff41f5096fb55367d1a225ce02b
google-deepmind · Feb 12, 2024 · 3019726 · 3019726
1 parent 382726e
commit 3019726
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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',