Fix transition_waiting_queued validation

dask · Feb 9, 2023 · b70bcc1 · b70bcc1
1 parent 33f191d
commit b70bcc1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -2663,7 +2663,14 @@ def transition_waiting_queued(self, key: str, stimulus_id: str) -> RecsMsgs:
         ts = self.tasks[key]
 
         if self.validate:
-            assert not self.idle_task_count, (ts, self.idle_task_count)
+            if self.is_rootish(ts):
+                assert not self.idle_task_count, (ts, self.idle_task_count)
+            else:
+                assert (
+                    ts.worker_restrictions
+                    or ts.host_restrictions
+                    or ts.resource_restrictions
+                )
             self._validate_ready(ts)
 
         ts.state = "queued"