Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions tests/benchmarks/test_work_stealing.py
Original file line number Diff line number Diff line change
@@ -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


Expand All @@ -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
):
Expand All @@ -30,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):
Expand Down