[Error]: I got unexpected error using enjoy() with pretrain model

[tiencapham]

[Error]: I got unexpected error using enjoy() with pretrain model

[tiencapham]

Silly me, but I solved it by:

pip install -e .[docs,tests,extra] in stable baseline.

Thanks for your help

[qgallouedec]

Hi, make sure to install atari envs.

pip install stable-baselines3[extra]

Answer in the doc: https://stable-baselines3.readthedocs.io/en/master/guide/examples.html#id2

[tiencapham]

Hi, make sure to install atari envs.

pip install stable-baselines3[extra]

Answer in the doc: https://stable-baselines3.readthedocs.io/en/master/guide/examples.html#id2

Hi,
Thank for your response.

I actually modified the stable-baseline3 and installed it from source. Maybe that is the reason I missed this step.

How can I install the atari envs in this case?

Best,
Tien Pham

[qgallouedec]

The answer is still in the doc, in the exact same paragraph :)

pip install gymnasium[atari, accept-rom-license]

Command out-of-date in the doc, fixed by DLR-RM/stable-baselines3#1773