fix(nyz): add constraint about numpy<2

ruiheng123 · Jun 20, 2024 · 91bc342 · 91bc342
1 parent 8f3f9eb
commit 91bc342
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -55,7 +55,7 @@
         'gym==0.25.1',  # pypy incompatible; some environments only support gym==0.22.0
         'gymnasium',
         'torch>=1.1.0',
-        'numpy>=1.18.0',
+        'numpy>=1.18.0,<2',
         'DI-treetensor>=0.4.0',
         'DI-toolkit>=0.1.0',
         'trueskill',