fix(ir): use rlz.map_to() rule instead of isin to normalize tempo…

…ral units
ibis-project · Feb 28, 2023 · a1c46a2 · a1c46a2
1 parent b3601c6
commit a1c46a2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ibis/expr/operations/temporal.py b/ibis/expr/operations/temporal.py
@@ -81,7 +81,7 @@ class TimestampUnary(Unary):
 @public
 class TimestampTruncate(Value):
     arg = rlz.timestamp
-    unit = rlz.isin(_timestamp_units)
+    unit = rlz.map_to(_timestamp_units)
 
     output_shape = rlz.shape_like("arg")
     output_dtype = dt.timestamp
@@ -90,7 +90,7 @@ class TimestampTruncate(Value):
 @public
 class DateTruncate(Value):
     arg = rlz.date
-    unit = rlz.isin(_date_units)
+    unit = rlz.map_to(_date_units)
 
     output_shape = rlz.shape_like("arg")
     output_dtype = dt.date
@@ -99,7 +99,7 @@ class DateTruncate(Value):
 @public
 class TimeTruncate(Value):
     arg = rlz.time
-    unit = rlz.isin(_time_units)
+    unit = rlz.map_to(_time_units)
 
     output_shape = rlz.shape_like("arg")
     output_dtype = dt.time