From 3ab468573f0e64dcee29630625408724fd2bb969 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Mon, 19 Sep 2022 10:34:37 -0500 Subject: [PATCH 1/2] Unskip test_work_stealing_on_scaling_up --- tests/benchmarks/test_work_stealing.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/benchmarks/test_work_stealing.py b/tests/benchmarks/test_work_stealing.py index dcb9c6bc05..382062324f 100644 --- a/tests/benchmarks/test_work_stealing.py +++ b/tests/benchmarks/test_work_stealing.py @@ -1,11 +1,13 @@ import time import dask.array as da +import distributed import numpy as np import pytest from coiled import Cluster from dask import delayed, utils from distributed import Client +from packaging.version import Version from tornado.ioloop import PeriodicCallback @@ -16,11 +18,10 @@ def test_trivial_workload_should_not_cause_work_stealing(small_client): small_client.gather(futs) -# @pytest.mark.xfail( -# distributed.__version__ == "2022.6.0", -# reason="https://github.com/dask/distributed/issues/6624", -# ) -@pytest.mark.skip("https://github.com/coiled/coiled-runtime/issues/336") +@pytest.mark.xfail( + Version(distributed.__version__) < Version("2022.6.1"), + reason="https://github.com/dask/distributed/issues/6624", +) def test_work_stealing_on_scaling_up( test_name_uuid, upload_cluster_dump, benchmark_all ): From 215a2b92e4fb077a21a47e16a85d5b7c94a5371d Mon Sep 17 00:00:00 2001 From: Ian Rose Date: Mon, 19 Sep 2022 13:43:40 -0700 Subject: [PATCH 2/2] Package sync --- tests/benchmarks/test_work_stealing.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/benchmarks/test_work_stealing.py b/tests/benchmarks/test_work_stealing.py index 382062324f..ae9fc47b66 100644 --- a/tests/benchmarks/test_work_stealing.py +++ b/tests/benchmarks/test_work_stealing.py @@ -31,6 +31,7 @@ def test_work_stealing_on_scaling_up( worker_vm_types=["t3.medium"], scheduler_vm_types=["t3.xlarge"], wait_for_workers=True, + package_sync=True, ) as cluster: with Client(cluster) as client: with upload_cluster_dump(client, cluster), benchmark_all(client):