Change total_occupancy directly

dask · Jan 5, 2021 · bec92c2 · bec92c2
1 parent 49db986
commit bec92c2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -6140,16 +6140,15 @@ def _reevaluate_occupancy_worker(self, ws: WorkerState):
         """ See reevaluate_occupancy """
         old: double = ws._occupancy
         new: double = 0
-        diff: double
         ts: TaskState
         est: double
         for ts in ws._processing:
             est = self.set_duration_estimate(ts, ws)
             new += est
 
         ws._occupancy = new
-        diff = new - old
-        self.total_occupancy += diff
+        self.total_occupancy -= old
+        self.total_occupancy += new
         self.check_idle_saturated(ws)
 
         # significant increase in duration