stable-baselines3/stable_baselines3/common
Mark Smith 9a749389d3
Cast learning_rate to float lambda for pickle safety when doing model.load (#1901)
* create failing test for unpickle error

* Fix learning_rate argument causing failure in weights_only=True if passed a function with non-float types

* Updated with feedback from araffin on PR#1901

* Update test and version

* Update changelog and SBX doc

---------

Co-authored-by: Antonin Raffin <antonin.raffin@ensta.org>
2024-04-22 10:04:01 +02:00
..
envs BitFlippingEnv argument check and docs clarification (#1698) 2023-09-27 10:18:30 +02:00
sb2_compat Fix type annotations of buffers (#1700) 2023-09-28 18:52:46 +02:00
vec_env Update to black style v24 (#1834) 2024-02-13 11:36:05 +01:00
__init__.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
atari_wrappers.py Fix VecEnv type hints (#1736) 2023-11-08 09:46:40 +01:00
base_class.py Fix docstring for `log_interval` to differentiate between on-policy/off-policy logging frequency (#1855) 2024-03-04 11:42:16 +01:00
buffers.py fix: Follow PEP8 guidelines and evaluate falsy to truthy with not rather than is False. (#1707) 2023-10-09 12:21:12 +02:00
callbacks.py Update dependencies (remove sphinx type hint plugin), protect type aliases 2023-10-23 20:14:15 +02:00
distributions.py Fix sum_independent_dims docstring to reflect output shape (#1851) 2024-02-27 14:49:42 +01:00
env_checker.py Update env checker for spaces with non-zero start (#1845) 2024-02-19 16:44:02 +01:00
env_util.py vec_envs fix seed() causing a reset (#1486) 2023-05-20 10:30:54 +02:00
evaluation.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
logger.py Fix reward of SimpleMultiObsEnv to always be float (#1676) 2023-09-16 08:56:04 +02:00
monitor.py Fix VecEnv type hints (#1736) 2023-11-08 09:46:40 +01:00
noise.py vec_envs fix seed() causing a reset (#1486) 2023-05-20 10:30:54 +02:00
off_policy_algorithm.py Fix VecEnv type hints (#1736) 2023-11-08 09:46:40 +01:00
on_policy_algorithm.py Log success rate for on policy algorithms (#1870) 2024-03-22 12:13:48 +01:00
policies.py Update ruff and documentation for hf sb3 (#1866) 2024-03-11 13:53:06 +01:00
preprocessing.py Fix policies type annotations (#1735) 2023-11-06 18:35:28 +01: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 Update ruff and documentation for hf sb3 (#1866) 2024-03-11 13:53:06 +01:00
torch_layers.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
type_aliases.py Update to black style v24 (#1834) 2024-02-13 11:36:05 +01:00
utils.py Cast learning_rate to float lambda for pickle safety when doing model.load (#1901) 2024-04-22 10:04:01 +02:00