From e54d3cf51f5d082bed01e385fc906455183ae88c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Quentin=20Gallou=C3=A9dec?=
 <45557362+qgallouedec@users.noreply.github.com>
Date: Fri, 28 Jun 2024 00:22:18 +0200
Subject: [PATCH] Numpy < 2.0 (#44)

* Update setup.py

* Update setup.py

* Update setup.py

* Update setup.py

* Update setup.py
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 95eb06e..997d683 100644
--- 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"]