Colab rl-baselines-zoo Error while finding module specification for 'train.py'

DLR-RM / rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

https://rl-baselines3-zoo.readthedocs.io

MIT License

1.99k stars 508 forks source link

Colab rl-baselines-zoo Error while finding module specification for 'train.py' #70

Closed lorepieri8 closed 3 years ago

lorepieri8 commented 3 years ago

Describe the bug /usr/bin/python3: Error while finding module specification for 'train.py' (ModuleNotFoundError: __path__ attribute not found on 'train' while trying to find 'train.py')

Code example Run the cell !python -m train.py --algo ppo --env MountainCar-v0 -n 50000 -optimize --n-trials 1000 --n-jobs 2 --sampler tpe --pruner median in https://colab.research.google.com/github/Stable-Baselines-Team/rl-colab-notebooks/blob/sb3/rl-baselines-zoo.ipynb

araffin commented 3 years ago

Hello, thanks for reporting the issue. Just remove the "-m" and it should work ;)

lorepieri8 commented 3 years ago

Yep, working now.

araffin commented 3 years ago

now fixed ;) (I pushed the new version)