diff --git a/python/tempo/ml.py b/python/tempo/ml.py
index 3db858af..67c73bf2 100644
--- 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)