From 67bca9ed78efaa22b830f3a8549be76129d231ad Mon Sep 17 00:00:00 2001 From: R7L208 <22798188+R7L208@users.noreply.github.com> Date: Wed, 10 Aug 2022 15:44:12 -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 6b7d462d..6a515b33 100644 --- 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)