Fix allowed types for save util (#1693)

DLR-RM · Sep 24, 2023 · b85fa75 · b85fa75
1 parent f4c5b1e
commit b85fa75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable_baselines3/common/save_util.py b/stable_baselines3/common/save_util.py
@@ -205,7 +205,7 @@ def open_path(
     """
     # Note(antonin): the true annotation should be IO[bytes]
     # but there is not easy way to check that
-    allowed_types = (io.BufferedWriter, io.BufferedReader, io.BytesIO)
+    allowed_types = (io.BufferedWriter, io.BufferedReader, io.BytesIO, io.BufferedRandom)
     if not isinstance(path, allowed_types):
         raise TypeError(f"Path {path} parameter has invalid type: expected one of {allowed_types}.")
     if path.closed: