update typing for extractStateIntervals

Author: R7L208

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