wip [skip ci]

pola-rs · Sep 10, 2024 · db5a9db · db5a9db
1 parent 35b254f
commit db5a9db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/py-polars/tests/unit/datatypes/test_list.py b/py-polars/tests/unit/datatypes/test_list.py
@@ -297,7 +297,7 @@ def test_flat_aggregation_to_list_conversion_6918() -> None:
 
     assert df.group_by("a", maintain_order=True).agg(
         pl.concat_list([pl.col("b").list.get(i).mean().implode() for i in range(2)])
-    ).to_dict(as_series=False) == {"a": [1, 2], "b": [[[0.0, 1.0]], [[3.0, 4.0]]]}
+    ).to_dict(as_series=False) == {"a": [1, 2], "b": [[0.0, 1.0], [3.0, 4.0]]}
 
 
 def test_list_count_matches() -> None:

diff --git a/py-polars/tests/unit/operations/aggregation/test_aggregations.py b/py-polars/tests/unit/operations/aggregation/test_aggregations.py
@@ -258,7 +258,7 @@ def test_err_on_implode_and_agg() -> None:
         pl.col("type").implode().list.head().alias("foo")
     ).to_dict(as_series=False) == {
         "type": ["water", "fire", "earth"],
-        "foo": [[["water", "water"]], [["fire"]], [["earth"]]],
+        "foo": [["water", "water"], ["fire"], ["earth"]],
     }
 
     # but not during a window function as the groups cannot be mapped back