stable-baselines3/stable_baselines3/common
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
..
sb2_compat Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
vec_env Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
__init__.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
atari_wrappers.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
base_class.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
bit_flipping_env.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
buffers.py Small docstring improvements related to the notion of Rollout (#206) 2020-11-02 11:45:08 +01:00
callbacks.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
cmd_util.py Fix off policy features extractor (#198) 2020-10-27 14:24:59 +01:00
distributions.py Fix off policy features extractor (#198) 2020-10-27 14:24:59 +01:00
env_checker.py add check to ensure action space is non-dict non-tuple for env_checker nan check (#192) 2020-10-19 00:23:51 +03:00
env_util.py Rename cmd_util to env_util (#197) 2020-10-22 11:05:52 +02:00
evaluation.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
identity_env.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
logger.py Add support to log videos via tensorboard (#196) 2020-10-22 11:33:58 +02:00
monitor.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
noise.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
off_policy_algorithm.py Small docstring improvements related to the notion of Rollout (#206) 2020-11-02 11:45:08 +01:00
on_policy_algorithm.py Small docstring improvements related to the notion of Rollout (#206) 2020-11-02 11:45:08 +01:00
policies.py Fix off policy features extractor (#198) 2020-10-27 14:24:59 +01:00
preprocessing.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
results_plotter.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
running_mean_std.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
save_util.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
torch_layers.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
type_aliases.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
utils.py Implement HER (#120) 2020-10-22 11:56:43 +02:00