Use typing.List rather than list

DLR-RM · Apr 3, 2024 · 2f403da · 2f403da
1 parent 39f0900
commit 2f403da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stable_baselines3/common/vec_env/vec_video_recorder.py b/stable_baselines3/common/vec_env/vec_video_recorder.py
@@ -53,7 +53,7 @@ def __init__(
         self.enabled = enabled
         self._closed = False
 
-        self.render_history: list[np.ndarray] = []
+        self.render_history: List[np.ndarray] = []
         self.env = env
 
         self.render_mode = env.render_mode
@@ -100,7 +100,7 @@ def __init__(
         self.write_metadata()
 
         logger.info(f"Starting new video recorder writing to {self.path}")
-        self.recorded_frames: list[np.ndarray] = []
+        self.recorded_frames: List[np.ndarray] = []
 
     @property
     def functional(self):