Fix test_merge_p2p_shuffle_reused_dataframe_with_different_parameters (…

…#8422)
dask · Dec 19, 2023 · 4d41d32 · 4d41d32
1 parent 25a1bfe
commit 4d41d32
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/distributed/shuffle/tests/test_merge.py b/distributed/shuffle/tests/test_merge.py
@@ -116,7 +116,9 @@ async def test_merge_p2p_shuffle_reused_dataframe_with_different_parameters(c, s
     out = (
         ddf1.merge(ddf2, left_on="a", right_on="x", shuffle="p2p")
         # Vary the number of output partitions for the shuffles of dd2
-        .repartition(20).merge(ddf2, left_on="b", right_on="x", shuffle="p2p")
+        .repartition(npartitions=20).merge(
+            ddf2, left_on="b", right_on="x", shuffle="p2p"
+        )
     )
     # Generate unique shuffle IDs if the input frame is the same but
     # parameters differ. Reusing shuffles in merges is dangerous because of the