Fix mypy error

DLR-RM · Jan 7, 2025 · 02dca85 · 02dca85
1 parent 57e8b97
commit 02dca85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/stable_baselines3/common/save_util.py b/stable_baselines3/common/save_util.py
@@ -181,7 +181,7 @@ def json_to_data(json_string: str, custom_objects: Optional[dict[str, Any]] = No
 @functools.singledispatch
 def open_path(
     path: Union[str, pathlib.Path, io.BufferedIOBase], mode: str, verbose: int = 0, suffix: Optional[str] = None
-) -> Union[io.BufferedWriter, io.BufferedReader, io.BytesIO]:
+) -> Union[io.BufferedWriter, io.BufferedReader, io.BytesIO, io.BufferedRandom]:
     """
     Opens a path for reading or writing with a preferred suffix and raises debug information.
     If the provided path is a derivative of io.BufferedIOBase it ensures that the file

diff --git a/tests/test_vec_envs.py b/tests/test_vec_envs.py
@@ -17,7 +17,7 @@
 from stable_baselines3.common.vec_env import DummyVecEnv, SubprocVecEnv, VecFrameStack, VecNormalize, VecVideoRecorder
 
 try:
-    import moviepy
+    import moviepy  # noqa: F401
 
     have_moviepy = True
 except ImportError: