Update to black style v24 (#1834)

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:
 ^^^^^^^^^^^^^^

setup.py

@@ -122,7 +122,7 @@ setup(
             # 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",

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

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
 

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
 

stable_baselines3/common/vec_env/vec_frame_stack.py

@@ -29,7 +29,12 @@ class VecFrameStack(VecEnvWrapper):
 
     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

stable_baselines3/version.txt

@@ -1 +1 @@
-2.3.0a1
+2.3.0a2