huggingface / huggingface_sb3

Additional code for Stable-baselines3 to load and upload models from the Hub.
77 stars 23 forks source link

Gymnasium support, with shimmy #34

Closed EdoardoPona closed 1 year ago

EdoardoPona commented 1 year ago

relevant to #30

IMPORTANT: some tests do not yet pass, due to loading policies from the hub that require gym

simoninithomas commented 1 year ago

Hey there 👋 we don't add shimmy by default because it's installed in SB3 or RL-Zoo by default.

We currently updating the huggingface_sb3 package you can find the PR here 👉 https://github.com/huggingface/huggingface_sb3/pull/35