feat(polars): add TimeFromHMS op

ibis-project · Dec 31, 2024 · acd3495 · acd3495
1 parent 56a66de
commit acd3495
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -1468,6 +1468,15 @@ def execute_timestamp_range(op, **kw):
     return pl.datetime_ranges(start, stop, f"{step}{unit}", closed="left")
 
 
+@translate.register(ops.TimeFromHMS)
+def execute_time_from_hms(op, **kw):
+    return pl.time(
+        hour=translate(op.hours, **kw),
+        minute=translate(op.minutes, **kw),
+        second=translate(op.seconds, **kw),
+    )
+
+
 @translate.register(ops.DropColumns)
 def execute_drop_columns(op, **kw):
     parent = translate(op.parent, **kw)

diff --git a/ibis/backends/tests/test_temporal.py b/ibis/backends/tests/test_temporal.py
@@ -1535,7 +1535,7 @@ def test_timestamp_with_timezone_literal(con, timezone, expected):
 
 
 @pytest.mark.notimpl(
-    ["datafusion", "pyspark", "polars", "mysql", "oracle", "databricks"],
+    ["datafusion", "pyspark", "mysql", "oracle", "databricks"],
     raises=com.OperationNotDefinedError,
 )
 @pytest.mark.notyet(