From 408d6a0c5b366f32e2f3aec95cec6ba7c02629f6 Mon Sep 17 00:00:00 2001 From: Adam Gleave Date: Thu, 7 Sep 2023 15:07:48 -0700 Subject: [PATCH] Remove HuggingFace pin --- setup.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 6b3d96cc4..5a1846697 100644 --- a/setup.py +++ b/setup.py @@ -207,10 +207,7 @@ def get_local_version(version: "ScmVersion", time_format="%Y%m%d") -> str: STABLE_BASELINES3, "sacred>=0.8.4", "tensorboard>=1.14", - # TODO(ernestum): remove once https://github.com/huggingface/huggingface_sb3/issues/37 is - # fixed - "huggingface_sb3==2.2.5", - "optuna>=3.0.1", + "huggingface_sb3~=2.3", "datasets>=2.8.0", ], tests_require=TESTS_REQUIRE,