From 1cba1bbd2f129f3e3140d6a1e478dd4b3979a2bf Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Tue, 13 Feb 2024 11:36:05 +0100
Subject: [PATCH] Update to black style v24 (#1834)

---
 docs/misc/changelog.rst                             | 3 ++-
 setup.py                                            | 2 +-
 stable_baselines3/common/save_util.py               | 1 +
 stable_baselines3/common/type_aliases.py            | 1 +
 stable_baselines3/common/vec_env/util.py            | 1 +
 stable_baselines3/common/vec_env/vec_frame_stack.py | 7 ++++++-
 stable_baselines3/version.txt                       | 2 +-
 7 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst
index 006f156..cf101af 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 817fae2..763a6a3 100644
--- a/setup.py
+++ b/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",
diff --git a/stable_baselines3/common/save_util.py b/stable_baselines3/common/save_util.py
index 0cbf6d4..9fca6a8 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 d75e115..85d0906 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 2a03d8e..855f50e 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 d412a96..daa2b36 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 @@ 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
diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt
index 4d04ad9..34109b6 100644
--- a/stable_baselines3/version.txt
+++ b/stable_baselines3/version.txt
@@ -1 +1 @@
-2.3.0a1
+2.3.0a2