fix(rust,python): return correct display/repr names for NaN-related e…

…xpressions (#6721)
pola-rs · Feb 8, 2023 · 5df0ce2 · 5df0ce2
1 parent 9ca2245
commit 5df0ce2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/polars/polars-lazy/polars-plan/src/dsl/function_expr/mod.rs b/polars/polars-lazy/polars-plan/src/dsl/function_expr/mod.rs
@@ -150,7 +150,11 @@ impl Display for FunctionExpr {
             #[cfg(all(feature = "rolling_window", feature = "moment"))]
             RollingSkew { .. } => "rolling_skew",
             ShiftAndFill { .. } => "shift_and_fill",
-            Nan(_) => "nan",
+            Nan(func) => match func {
+                NanFunction::IsNan => "is_nan",
+                NanFunction::IsNotNan => "is_not_nan",
+                NanFunction::DropNans => "drop_nans",
+            },
             #[cfg(feature = "round_series")]
             Clip { min, max } => match (min, max) {
                 (Some(_), Some(_)) => "clip",