search

araffin / rl-baselines-zoo

A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.
https://stable-baselines.readthedocs.io/
MIT License
1.12k stars 206 forks source link

SB3 v1.1 Breaking changes #116

Closed jkterry1 closed 3 years ago

jkterry1 commented 3 years ago

I recently upgraded to a SB3 1.1 prerelease (what's currently master), and the breaking changes have minor implications for RL zoo. I don't know if you want PRs for this now, for me to wait until 1.1 is properly released, or if you want to do the upgrade yourself what. So far I've found one problem I've had to fix on my fork though:

Traceback (most recent call last):
  File "train.py", line 15, in <module>
    from utils.exp_manager import ExperimentManager
  File "/home/justin_terry/rl-baselines3-zoo/utils/exp_manager.py", line 26, in <module>
    from stable_baselines3.common.vec_env.obs_dict_wrapper import ObsDictWrapper
ModuleNotFoundError: No module named 'stable_baselines3.common.vec_env.obs_dict_wrapper'
jkterry1 commented 3 years ago

I accidentally created this on the old repo, my apologies.