From 4d41d326b3aa63db3c3861d7fd2676c77af42207 Mon Sep 17 00:00:00 2001 From: Hendrik Makait Date: Tue, 19 Dec 2023 13:48:42 +0100 Subject: [PATCH] Fix test_merge_p2p_shuffle_reused_dataframe_with_different_parameters (#8422) --- distributed/shuffle/tests/test_merge.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/distributed/shuffle/tests/test_merge.py b/distributed/shuffle/tests/test_merge.py index 7ed6e942c31..08d714e77af 100644 --- 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