Update dask_sql/physical/rex/core/call.py

Co-authored-by: Ayush Dattagupta <[email protected]>
dask-contrib · Oct 13, 2022 · 072b5a0 · 072b5a0
1 parent 198b510
commit 072b5a0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/dask_sql/physical/rex/core/call.py b/dask_sql/physical/rex/core/call.py
@@ -44,9 +44,7 @@ def as_timelike(op):
         return np.timedelta64(op, "D")
     elif isinstance(op, str):
         return np.datetime64(op)
-    elif pd.api.types.is_datetime64_dtype(op):
-        return op
-    elif isinstance(op, np.timedelta64):
+    elif pd.api.types.is_datetime64_dtype(op) or isinstance(op, np.timedelta64):
         return op
     else:
         raise ValueError(f"Don't know how to make {type(op)} timelike")