Update setup.py

huggingface · Jun 27, 2024 · 8cbd895 · 8cbd895
1 parent bd67904
commit 8cbd895
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
     "flake8>=3.8.3",
 ]
 
-extras["test"] = ["pytest"]
+extras["test"] = ["pytest", "gymnasium[classic-control"]
 
 extras["dev"] = extras["quality"] + extras["test"]