Update to black style v24 (#1834)

DLR-RM · Feb 13, 2024 · 1cba1bb · 1cba1bb
1 parent beee427
commit 1cba1bb
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -1 +1 @@
-2.3.0a1
+2.3.0a2