update typing for extractStateIntervals

databrickslabs · Aug 10, 2022 · 67bca9e · 67bca9e
1 parent 6f795b9
commit 67bca9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tempo/tsdf.py b/python/tempo/tsdf.py
@@ -1362,7 +1362,7 @@ def extractStateIntervals(
         """
 
         # https://spark.apache.org/docs/latest/sql-ref-null-semantics.html#comparison-operators-
-        def null_safe_equals(col1: Column, col2: Column) -> Column[bool]:
+        def null_safe_equals(col1: Column, col2: Column) -> Column:
             return (
                 f.when(col1.isNull() & col2.isNull(), True)
                 .when(col1.isNull() | col2.isNull(), False)