stable-baselines3/stable_baselines3/common
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
..
envs Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
sb2_compat Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
vec_env Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
__init__.py
atari_wrappers.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
base_class.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
buffers.py Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
callbacks.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
distributions.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
env_checker.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
env_util.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
evaluation.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
logger.py Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
monitor.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
noise.py Add the argument dtype (default to float32) to the noise (#1301) 2023-02-07 13:42:14 +01:00
off_policy_algorithm.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
on_policy_algorithm.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
policies.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
preprocessing.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
results_plotter.py Fix Atari Roms download, enable RUF linting (#1379) 2023-03-12 18:47:52 +01:00
running_mean_std.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
save_util.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
torch_layers.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
type_aliases.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
utils.py Update issue templates and env infos (#1451) 2023-04-14 13:50:14 +02:00