Skip to content

Commit a8206a8

Browse files
committed
add some checks
1 parent aaab679 commit a8206a8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

services/dask-sidecar/tests/unit/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ def local_cluster(app_environment: EnvVarsDict) -> Iterator[distributed.LocalClu
121121
) as cluster:
122122
assert cluster
123123
assert isinstance(cluster, distributed.LocalCluster)
124+
print(cluster.workers)
124125
yield cluster
125126

126127

@@ -129,6 +130,7 @@ def dask_client(
129130
local_cluster: distributed.LocalCluster,
130131
) -> Iterator[distributed.Client]:
131132
with distributed.Client(local_cluster) as client:
133+
client.wait_for_workers(1, timeout=10)
132134
yield client
133135

134136

0 commit comments

Comments
 (0)