stable-baselines3/stable_baselines3/common/vec_env
Anssi e2b6f5460f
Avoid transposing channel-first envs (#213)
* Add test for channel-first environments

* Add support for channel-first envs, including more tests

* Update changelog

* Run black

* Run black, again

* Improve NatureCNN error message

* Update image checks and FrameStack wrapper

* Update tests

* Update docs

* Run isort

* Reformat

* Fixes: avoid breaking changes for non-image env

* Add additional checks

* Update docstring

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2020-11-03 12:34:09 +01:00
..
__init__.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
base_vec_env.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
dummy_vec_env.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
obs_dict_wrapper.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
subproc_vec_env.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
util.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
vec_check_nan.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
vec_frame_stack.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
vec_normalize.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
vec_transpose.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
vec_video_recorder.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00