c

pola-rs · Jan 16, 2025 · 14f631a · 14f631a
1 parent 5435efd
commit 14f631a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py-polars/tests/unit/operations/map/test_map_groups.py b/py-polars/tests/unit/operations/map/test_map_groups.py
@@ -162,6 +162,6 @@ def foo(x: pl.Series) -> pl.Series:
         pl.DataFrame({"key": [0, 0, 1], "a": [None, None, None]})
         .group_by("key")
         .agg(pl.map_groups(exprs=["a"], function=foo))  # type: ignore[arg-type]
-        .sort("a"),
+        .sort("key"),
         pl.DataFrame({"key": [0, 1], "a": [None, None]}),
     )