diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index 006f156ef..cf101af45 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -3,7 +3,7 @@ Changelog ========== -Release 2.3.0a1 (WIP) +Release 2.3.0a2 (WIP) -------------------------- Breaking Changes: @@ -54,6 +54,7 @@ Deprecations: Others: ^^^^^^^ +- Updated black from v23 to v24 Documentation: ^^^^^^^^^^^^^^ diff --git a/setup.py b/setup.py index 817fae22a..763a6a376 100644 --- a/setup.py +++ b/setup.py @@ -122,7 +122,7 @@ # Lint code and sort imports (flake8 and isort replacement) "ruff>=0.0.288", # Reformat - "black>=23.9.1,<24", + "black>=24.2.0,<25", ], "docs": [ "sphinx>=5,<8", diff --git a/stable_baselines3/common/save_util.py b/stable_baselines3/common/save_util.py index 0cbf6d4e2..9fca6a832 100644 --- a/stable_baselines3/common/save_util.py +++ b/stable_baselines3/common/save_util.py @@ -2,6 +2,7 @@ Save util taken from stable_baselines used to serialize data (class parameters) of model classes """ + import base64 import functools import io diff --git a/stable_baselines3/common/type_aliases.py b/stable_baselines3/common/type_aliases.py index d75e11531..85d09066e 100644 --- a/stable_baselines3/common/type_aliases.py +++ b/stable_baselines3/common/type_aliases.py @@ -1,4 +1,5 @@ """Common aliases for type hints""" + from enum import Enum from typing import TYPE_CHECKING, Any, Callable, Dict, List, NamedTuple, Optional, Protocol, SupportsFloat, Tuple, Union diff --git a/stable_baselines3/common/vec_env/util.py b/stable_baselines3/common/vec_env/util.py index 2a03d8e70..855f50edc 100644 --- a/stable_baselines3/common/vec_env/util.py +++ b/stable_baselines3/common/vec_env/util.py @@ -1,6 +1,7 @@ """ Helpers for dealing with vectorized environments. """ + from collections import OrderedDict from typing import Any, Dict, List, Tuple diff --git a/stable_baselines3/common/vec_env/vec_frame_stack.py b/stable_baselines3/common/vec_env/vec_frame_stack.py index d412a96a2..daa2b365c 100644 --- a/stable_baselines3/common/vec_env/vec_frame_stack.py +++ b/stable_baselines3/common/vec_env/vec_frame_stack.py @@ -29,7 +29,12 @@ def __init__(self, venv: VecEnv, n_stack: int, channels_order: Optional[Union[st def step_wait( self, - ) -> Tuple[Union[np.ndarray, Dict[str, np.ndarray]], np.ndarray, np.ndarray, List[Dict[str, Any]],]: + ) -> Tuple[ + Union[np.ndarray, Dict[str, np.ndarray]], + np.ndarray, + np.ndarray, + List[Dict[str, Any]], + ]: observations, rewards, dones, infos = self.venv.step_wait() observations, infos = self.stacked_obs.update(observations, dones, infos) # type: ignore[arg-type] return observations, rewards, dones, infos diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt index 4d04ad95c..34109b68e 100644 --- a/stable_baselines3/version.txt +++ b/stable_baselines3/version.txt @@ -1 +1 @@ -2.3.0a1 +2.3.0a2