review comments

dask · Oct 13, 2021 · bb3899b · bb3899b
1 parent ac2b153
commit bb3899b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -2765,10 +2765,10 @@ def transition_processing_memory(
 
             s: set = self._unknown_durations.pop(ts._prefix._name, set())
             tts: TaskState
+            steal = self.extensions.get("stealing")
             for tts in s:
                 if tts._processing_on:
                     self.set_duration_estimate(tts, tts._processing_on)
-                    steal = self.extensions.get("stealing")
                     if steal:
                         steal.put_key_in_stealable(tts)
 

diff --git a/distributed/stealing.py b/distributed/stealing.py
@@ -172,8 +172,6 @@ def move_task_request(self, ts, victim, thief) -> str:
             if ts in self.in_flight:
                 return "in-flight"
             stimulus_id = f"steal-{time()}"
-            if self.scheduler.validate:
-                assert victim is ts.processing_on
 
             key = ts.key
             self.remove_key_from_stealable(ts)