Add huggingface_sb3<=2.2.5 version requirement

HumanCompatibleAI · Sep 5, 2023 · fc35e90 · fc35e90
1 parent 5b0b531
commit fc35e90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -207,7 +207,7 @@ def get_local_version(version: "ScmVersion", time_format="%Y%m%d") -> str:
         STABLE_BASELINES3,
         "sacred>=0.8.4",
         "tensorboard>=1.14",
-        "huggingface_sb3>=2.2.1",
+        "huggingface_sb3>=2.2.1,<=2.2.5",
         "datasets>=2.8.0",
     ],
     tests_require=TESTS_REQUIRE,