No randomness if < 10 workers to choose from

Fixes flaky `test_balance_with_restrictions`
dask · Jun 21, 2021 · fcb165e · fcb165e
1 parent 13975cb
commit fcb165e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -7501,6 +7501,8 @@ def decide_worker(
             sample_from = idle_workers or worker_pool
             candidates.update(
                 random.choices(list(sample_from), k=min(10, len(sample_from)))
+                if len(sample_from) > 10
+                else sample_from
             )
     if valid_workers is None:
         if not candidates: