From db0e7a34c168dc13619842964b6630b7309a780c Mon Sep 17 00:00:00 2001 From: Hendrik Makait Date: Thu, 24 Oct 2024 19:37:30 +0200 Subject: [PATCH] Fix tests --- distributed/shuffle/tests/test_shuffle.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/distributed/shuffle/tests/test_shuffle.py b/distributed/shuffle/tests/test_shuffle.py index 894cf9ff127..5bc3c620898 100644 --- a/distributed/shuffle/tests/test_shuffle.py +++ b/distributed/shuffle/tests/test_shuffle.py @@ -2744,7 +2744,7 @@ async def test_flaky_connect_fails_without_retry(c, s, a, b): with mock.patch.object(a, "rpc", rpc): with raises_with_cause( expected_exception=RuntimeError, - match="P2P shuffling.*transfer", + match="P2P.*transfer", expected_cause=OSError, match_cause=None, ): @@ -2899,7 +2899,7 @@ def data_gen(): with raises_with_cause( RuntimeError, - r"shuffling \w* failed", + r"P2P \w* failed", ValueError, "meta", ):