Update scheduler.py

dask · Dec 18, 2024 · da45cb5 · da45cb5
1 parent 3bf8576
commit da45cb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -3082,7 +3082,7 @@ def is_rootish(self, ts: TaskState) -> bool:
         to override this heuristic.
         """
         # Check explicitly marked IO tasks
-        if ts.run_spec and ts.run_spec.io_task:
+        if ts.run_spec and ts.run_spec.data_producer_task:
             return True
 
         if ts.resource_restrictions or ts.worker_restrictions or ts.host_restrictions: