From 6f795b95428ac08351badace4191803a1e7a84ea Mon Sep 17 00:00:00 2001
From: R7L208 <22798188+R7L208@users.noreply.github.com>
Date: Wed, 10 Aug 2022 15:43:48 -0400
Subject: [PATCH] update typing for extractStateIntervals

---
 python/tempo/tsdf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tempo/tsdf.py b/python/tempo/tsdf.py
index 1b481657..6b7d462d 100644
--- 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