stable-baselines3/tests
Antonin RAFFIN d228364ccf
Add timeout handling for on-policy algorithms (#658)
* Add timeout handling for on-policy algorithms

* Fixes

* Fix infinite loop in eval

* Skip type check for python 3.9

* Fix for discrete obs + add docstring

* Fix A2C test

* Removed unused helper

* Add test for infinite horizon

* typed ast should be fixed

* Apply suggestions from code review

Co-authored-by: Anssi <kaneran21@hotmail.com>

Co-authored-by: Anssi <kaneran21@hotmail.com>
2021-11-16 17:19:16 +01:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_callbacks.py Doc fix and improve error messages (#598) 2021-10-08 18:08:31 +02:00
test_cnn.py Documentation update (#450) 2021-05-23 13:13:11 +02:00
test_custom_policy.py Fix discrete obs support (#296) 2021-01-21 02:42:33 +02:00
test_deterministic.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_dict_env.py Add timeout handling for on-policy algorithms (#658) 2021-11-16 17:19:16 +01:00
test_distributions.py Add method predict_values for ActorCriticPolicy (#569) 2021-09-15 14:03:04 +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 Dictionary Observations (#243) 2021-05-11 12:29:30 +02:00
test_gae.py Add timeout handling for on-policy algorithms (#658) 2021-11-16 17:19:16 +01:00
test_her.py Dictionary Observations (#243) 2021-05-11 12:29:30 +02:00
test_identity.py Add timeout handling for on-policy algorithms (#658) 2021-11-16 17:19:16 +01:00
test_logger.py Relax test 2021-10-31 19:03:28 +01:00
test_monitor.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_predict.py Fixes in is_vectorized_observation (#587) 2021-09-28 21:57:49 +02:00
test_run.py Fix train_freq at load time (#332) 2021-02-27 19:53:13 +01:00
test_save_load.py Fix set_env to keep the number of timesteps (#615) 2021-10-23 16:36:40 +02:00
test_sde.py Remove sde_net_arch + Simplify policy (#584) 2021-09-28 22:32:54 +03:00
test_spaces.py Fix discrete obs support (#296) 2021-01-21 02:42:33 +02:00
test_tensorboard.py Auto-formatting with black and isort (#97) 2020-07-16 16:12:16 +02:00
test_train_eval_mode.py Avoid putting target networks into training mode (#553) 2021-08-30 17:42:41 +02:00
test_utils.py System info helper (#613) 2021-10-18 10:43:56 +02: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 Use Monitor episode reward/length for evaluate_policy (#220) 2020-11-16 11:52:28 +01:00
test_vec_extract_dict_obs.py Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
test_vec_monitor.py Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
test_vec_normalize.py Fix set_env when using VecNormalize (#638) 2021-11-02 13:52:26 +02:00