Fix tests

dask · Oct 24, 2024 · db0e7a3 · db0e7a3
1 parent 9e8a0a4
commit db0e7a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",
     ):