stable-baselines3/tests
Antonin RAFFIN 63a0bb9da1
Type annotation bundle (logger, vec env, custom envs) (#1479)
* Switch from List to Sequence for `seed()` type hint

* Fix logger type hints

* Improve replay buffer type hints

* Fix custom envs type annotations

* Fix VecMonitor type hints

* Fix RMSprop type hint

* Fix vec extract dict obs type hints

* Fix vec frame stack type annotations

* Fix base vec env type hints

* Fix dummy vec env type hints

* Fix for mypy

* Fixes for the tests

* mypy doesn't like when we overwrite type

* fix step of SimpleMultiObsEnv

* remove useless type specification

* Rm useless type hint

* Improve logger type hint

* format

* rm useless type hint

* Re-add variables in constructor, remove unused import

---------

Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>
2023-05-04 20:27:15 +02:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_buffers.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_callbacks.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_cnn.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_custom_policy.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
test_deterministic.py Add the argument dtype (default to float32) to the noise (#1301) 2023-02-07 13:42:14 +01:00
test_dict_env.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_distributions.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_env_checker.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_envs.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_gae.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_her.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_identity.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_logger.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_monitor.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_predict.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_preprocessing.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_run.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_save_load.py Fix DQN target update interval for multi-env (#1463) 2023-04-27 18:35:33 +02:00
test_sde.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_spaces.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_tensorboard.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
test_train_eval_mode.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_utils.py Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
test_vec_check_nan.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_vec_envs.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_vec_extract_dict_obs.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_vec_monitor.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_vec_normalize.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
test_vec_stacked_obs.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00