refactor(rust): Add missing implicit datetime alias in ExprIR (#18809)

pola-rs · Sep 18, 2024 · 8114b52 · 8114b52
1 parent ffc7a70
commit 8114b52
Showing 1 changed file with 18 additions and 14 deletions.
diff --git a/crates/polars-plan/src/dsl/functions/temporal.rs b/crates/polars-plan/src/dsl/functions/temporal.rs
@@ -208,21 +208,25 @@ pub fn datetime(args: DatetimeArgs) -> Expr {
         ambiguous,
     ];
 
-    Expr::Function {
-        input,
-        function: FunctionExpr::TemporalExpr(TemporalFunction::DatetimeFunction {
-            time_unit,
-            time_zone,
+    Expr::Alias(
+        Arc::new(Expr::Function {
+            input,
+            function: FunctionExpr::TemporalExpr(TemporalFunction::DatetimeFunction {
+                time_unit,
+                time_zone,
+            }),
+            options: FunctionOptions {
+                collect_groups: ApplyOptions::ElementWise,
+                flags: FunctionFlags::default()
+                    | FunctionFlags::INPUT_WILDCARD_EXPANSION
+                    | FunctionFlags::ALLOW_RENAME,
+                fmt_str: "datetime",
+                ..Default::default()
+            },
         }),
-        options: FunctionOptions {
-            collect_groups: ApplyOptions::ElementWise,
-            flags: FunctionFlags::default()
-                | FunctionFlags::INPUT_WILDCARD_EXPANSION
-                | FunctionFlags::ALLOW_RENAME,
-            fmt_str: "datetime",
-            ..Default::default()
-        },
-    }
+        // TODO: follow left-hand rule in Polars 2.0.
+        PlSmallStr::from_static("datetime"),
+    )
 }
 
 /// Arguments used by `duration` in order to produce an [`Expr`] of [`Duration`]