Typecheck

TimeEval · Mar 6, 2024 · 3697c3c · 3697c3c
1 parent 3660821
commit 3697c3c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_generator/test_lengths.py b/tests/test_generator/test_lengths.py
@@ -30,17 +30,17 @@ def _update_config(
         self, length: int, include_creeping: bool = False
     ) -> Dict[str, Any]:
         config = self.config.copy()
-        config["timeseries"][0]["anomalies"][0]["length"] = length
+        config["timeseries"][0]["anomalies"][0]["length"] = length  # type: ignore
         cl = int(round(length * self.creeping_factor)) if include_creeping else 0
-        config["timeseries"][0]["anomalies"][0]["creeping-length"] = cl
+        config["timeseries"][0]["anomalies"][0]["creeping-length"] = cl  # type: ignore
         return config
 
     def _run_and_validate(self, config: Dict[str, Any]) -> None:
         gutentag = GutenTAG(seed=self.seed)
         gutentag.load_config_dict(config)
         ts = gutentag.generate(return_timeseries=True)
         self.assertIsNotNone(ts)
-        self.assertEqual(len(ts), 1)
+        self.assertEqual(len(ts), 1)  # type: ignore
 
     def test_lengths(self) -> None:
         for i in range(25, 50):