Compatibility fix for wandb and numpy version conflict.

opendilab · Jun 17, 2024 · 31eaa6a · 31eaa6a
1 parent 07bcd44
commit 31eaa6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
         exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
     install_requires=[
         'gym',
-        'numpy',
+        'numpy<=1.26.4',
         'torch>=2.2.0',
         'opencv-python',
         'tensordict',