From 48a8389b340cebbfd2be2bf911e64114fdfbb99a Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Tue, 7 Jan 2025 11:27:02 +0100 Subject: [PATCH] Fix mypy error --- stable_baselines3/common/save_util.py | 2 +- tests/test_vec_envs.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/stable_baselines3/common/save_util.py b/stable_baselines3/common/save_util.py index a85c9c2..3ce6601 100644 --- 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 index 0dc3404..a390145 100644 --- a/tests/test_vec_envs.py +++ b/tests/test_vec_envs.py @@ -17,7 +17,7 @@ from stable_baselines3.common.monitor import Monitor from stable_baselines3.common.vec_env import DummyVecEnv, SubprocVecEnv, VecFrameStack, VecNormalize, VecVideoRecorder try: - import moviepy + import moviepy # noqa: F401 have_moviepy = True except ImportError: