Workaround python/mypy#1153

toslunar · Mar 11, 2022 · 0c21181 · 0c21181
1 parent afff3d5
commit 0c21181
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pfrl/wrappers/monitor.py b/pfrl/wrappers/monitor.py
@@ -9,16 +9,14 @@ class _Stub:
         def __init__(self, *args, **kwargs):
             raise RuntimeError("Monitor is not available in this version of gym")
 
-    class _GymMonitor(_Stub):
+    class _GymMonitor(_Stub):  # type: ignore
         pass
 
     class _GymStatsRecorder(_Stub):
         pass
 
 else:
-    from gym.wrappers.monitoring.stats_recorder import (
-        StatsRecorder as _GymStatsRecorder,
-    )
+    from gym.wrappers.monitoring.stats_recorder import StatsRecorder as _GymStatsRecorder  # type: ignore
 
 
 class Monitor(_GymMonitor):