update typing for extractStateIntervals

databrickslabs · Aug 10, 2022 · 6f795b9 · 6f795b9
1 parent 620bcff
commit 6f795b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tempo/tsdf.py b/python/tempo/tsdf.py
@@ -1341,7 +1341,7 @@ def tempo_fourier_util(pdf):
     def extractStateIntervals(
         self,
         *metric_cols: str,
-        state_definition: Union[str, Callable[[Column, Column], Column[bool]]] = "=",
+        state_definition: Union[str, Callable[[Column, Column], Column]] = "=",
     ) -> DataFrame:
         """
         Extracts intervals from a :class:`~tsdf.TSDF` based on some notion of "state", as defined by the :param