Numpy < 2.0 (#44)

* Update setup.py * Update setup.py * Update setup.py * Update setup.py * Update setup.py
huggingface · Jun 27, 2024 · e54d3cf · e54d3cf
1 parent bd67904
commit e54d3cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
     "huggingface_hub>=0.21",
     "pyyaml~=6.0",
     "wasabi",
-    "numpy",
+    "numpy<2.0",
     "cloudpickle>=1.6",
     "stable-baselines3",
     "moviepy",
@@ -18,7 +18,7 @@
     "flake8>=3.8.3",
 ]
 
-extras["test"] = ["pytest"]
+extras["test"] = ["pytest", "gymnasium[classic-control]"]
 
 extras["dev"] = extras["quality"] + extras["test"]