fix test (#3222)

pola-rs · Apr 23, 2022 · 9e4145f · 9e4145f
1 parent fbd4824
commit 9e4145f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/py-polars/tests/test_queries.py b/py-polars/tests/test_queries.py
@@ -58,7 +58,9 @@ def test_agg_after_head() -> None:
     expected = pl.DataFrame({"a": [1, 2, 3], "b": [6, 9, 21]})
 
     for maintain_order in [True, False]:
-        out = df.groupby("a", maintain_order=True).agg([pl.col("b").head(3).sum()])
+        out = df.groupby("a", maintain_order=maintain_order).agg(
+            [pl.col("b").head(3).sum()]
+        )
 
         if not maintain_order:
             out = out.sort("a")