fix: Fix empty scalar agg type (#20051)

pola-rs · Nov 28, 2024 · 74d059f · 74d059f
1 parent 9f3f012
commit 74d059f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/crates/polars-core/src/frame/column/mod.rs b/crates/polars-core/src/frame/column/mod.rs
@@ -605,7 +605,7 @@ impl Column {
             Column::Partitioned(s) => series_agg(s.as_materialized_series(), groups).into_column(),
             Column::Scalar(s) => {
                 if s.is_empty() {
-                    return self.clone();
+                    return series_agg(s.as_materialized_series(), groups).into_column();
                 }
 
                 // We utilize the aggregation on Series to see:

diff --git a/py-polars/tests/unit/test_schema.py b/py-polars/tests/unit/test_schema.py
@@ -336,3 +336,12 @@ def test_raise_subnodes_18787() -> None:
                 pl.first().struct.field("a", "b").filter(pl.col("foo") == 1)
             )
         )
+
+
+def test_scalar_agg_schema_20044() -> None:
+    assert (
+        pl.DataFrame(None, schema={"a": pl.Int64, "b": pl.String, "c": pl.String})
+        .with_columns(d=pl.col("a").max())
+        .group_by("c")
+        .agg(pl.col("d").mean())
+    ).schema == pl.Schema([("c", pl.String), ("d", pl.Float64)])