fixing type-check issue

databrickslabs · Oct 31, 2024 · 1f16859 · 1f16859
1 parent 11f51cd
commit 1f16859
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tempo/ml.py b/python/tempo/ml.py
@@ -1,4 +1,4 @@
-from typing import List, Tuple
+from typing import Any, List, Tuple
 from functools import reduce
 
 from pyspark.sql import DataFrame
@@ -39,7 +39,7 @@ def __init__(
         timeSeriesCol: str = "event_ts",
         seriesIdCols: List[str] = [],
         gap: int = 0,
-        **other_kwargs
+        **other_kwargs: Any
     ) -> None:
         super(TimeSeriesCrossValidator, self).__init__(**other_kwargs)
         self._setDefault(timeSeriesCol="event_ts", seriesIdCols=[], gap=0)