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

Optional keyword argument to pass to the env constructor. Solve #58 #73

Closed caburu closed 4 years ago

caburu commented 4 years ago

I coded the necessary changes to allow users of the RL Baseline Zoo to pass additional parameters for the environment's constructor.

This is useful when using customized environments. Ideally the environments should be fixed, but especially during development phase it's very helpful to pass additional optional parameters. This solves the question #58. closes #58 .

For the parameter name ( env_kwargs) I chose the same as in stable_baselines.common.cmd_util.make_vec_env.

I've used the tests of the repository and it seems all be ok.