Fix Gymnasium version

DLR-RM · Mar 29, 2023 · 39eae4e · 39eae4e
1 parent 82e3128
commit 39eae4e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -102,8 +102,7 @@
     packages=[package for package in find_packages() if package.startswith("stable_baselines3")],
     package_data={"stable_baselines3": ["py.typed", "version.txt"]},
     install_requires=[
-        # TODO(antonin): update to point to a release number once it is merged
-        "gymnasium @ git+https://github.com/Farama-Foundation/Gymnasium@main",
+        "gymnasium==0.28.1",
         "numpy",
         "torch>=1.11",
         'typing_extensions>=4.0,<5; python_version < "3.8.0"',