search

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.9k stars 495 forks source link

[Feature Request] Add the `--gym-packages` flag to `record_video` #396

Open Quentin18 opened 10 months ago

Quentin18 commented 10 months ago

🚀 Feature

Add the --gym-packages flag to the record_video script in order to import external Gym environment package modules, similarly to the train and enjoy scripts.

Motivation

Actually, we can use the train and enjoy scripts with custom Gym environments, but we cannot use record_video.

Pitch

No response

Alternatives

No response

Additional context

No response

Checklist

araffin commented 10 months ago

Related: https://github.com/DLR-RM/rl-baselines3-zoo/issues/390 (might be already working)