Fix setter error as per python/mypy#5936

HumanCompatibleAI · Sep 7, 2022 · 479b2f1 · 479b2f1
1 parent 3171a8b
commit 479b2f1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/imitation/algorithms/preference_comparisons.py b/src/imitation/algorithms/preference_comparisons.py
@@ -302,8 +302,11 @@ def sample(self, steps: int) -> Sequence[types.TrajectoryWithRew]:
             trajectories.extend(list(exploration_trajs))
         return trajectories
 
-    # Type ignore due to https://github.com/python/mypy/issues/5936
-    @TrajectoryGenerator.logger.setter  # type: ignore[attr-defined]
+    @property
+    def logger(self):
+        return super().logger
+
+    @logger.setter
     def logger(self, value: imit_logger.HierarchicalLogger):
         self._logger = value
         self.algorithm.set_logger(self.logger)