stable-baselines3/tests
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 Init: TD3 2019-09-05 17:29:41 +02:00
test_callbacks.py Rename cmd_util to env_util (#197) 2020-10-22 11:05:52 +02:00
test_cnn.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
test_custom_policy.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
test_deterministic.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_distributions.py Update black version + update docker image (#151) 2020-08-27 23:02:59 +02:00
test_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
test_envs.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_her.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
test_identity.py Fix off-by-one GAE computation (#185) 2020-10-13 00:10:54 +03:00
test_logger.py Add support to log videos via tensorboard (#196) 2020-10-22 11:33:58 +02:00
test_monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_predict.py Allow to set a device when loading a model (#154) 2020-09-20 19:13:18 +02:00
test_run.py Fix off policy features extractor (#198) 2020-10-27 14:24:59 +01:00
test_save_load.py Fix env loading (#203) 2020-10-27 23:12:52 +02:00
test_sde.py Implement HER (#120) 2020-10-22 11:56:43 +02:00
test_spaces.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_tensorboard.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_utils.py Fix off policy features extractor (#198) 2020-10-27 14:24:59 +01:00
test_vec_check_nan.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_vec_envs.py Avoid transposing channel-first envs (#213) 2020-11-03 12:34:09 +01:00
test_vec_normalize.py Implement HER (#120) 2020-10-22 11:56:43 +02:00