stable-baselines3/stable_baselines3
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
..
a2c Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
common Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00
ddpg Upgrade black formatting (#1310) 2023-02-02 11:58:41 +01:00
dqn Fix DQN target update interval for multi-env (#1463) 2023-04-27 18:35:33 +02:00
her Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
ppo Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
sac Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
td3 Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
__init__.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
py.typed Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
version.txt Type annotation bundle (logger, vec env, custom envs) (#1479) 2023-05-04 20:27:15 +02:00