polish

Author: crusaderky

distributed/worker.py

@@ -2047,14 +2047,8 @@ def transition_missing_released(
     def transition_flight_missing(
         self, ts: TaskState, *, stimulus_id: str
     ) -> RecsInstrs:
-        if self.validate:
-            assert ts.done
-            assert not ts.who_has
-
-        ts.state = "missing"
-        self._missing_dep_flight.add(ts)
-        ts.done = False
-        return {}, []
+        assert ts.done
+        return self.transition_generic_missing(ts, stimulus_id=stimulus_id)
 
     def transition_generic_missing(
         self, ts: TaskState, *, stimulus_id: str