Remove stimulus_id from long running messages

distributed/scheduler.py

@@ -5713,18 +5713,13 @@ def release_worker_data(self, key, worker, stimulus_id):
         if recommendations:
             self.transitions(recommendations, stimulus_id)
 
-    def handle_long_running(
-        self, key=None, worker=None, compute_duration=None, stimulus_id=None
-    ):
+    def handle_long_running(self, key=None, worker=None, compute_duration=None):
         """A task has seceded from the thread pool
 
         We stop the task from being stolen in the future, and change task
         duration accounting as if the task has stopped.
         """
         parent: SchedulerState = cast(SchedulerState, self)
-        # TODO(sjperkins): This stimulus isn't passed anywhere, so
-        # could probably be removed from the signature before the PR is merged
-        assert stimulus_id
         if key not in parent._tasks:
             logger.debug("Skipping long_running since key %s was already released", key)
             return

distributed/worker.py

@@ -2451,13 +2451,7 @@ def transition_executing_long_running(
         return merge_recs_instructions(
             (
                 {},
-                [
-                    LongRunningMsg(
-                        key=ts.key,
-                        compute_duration=compute_duration,
-                        stimulus_id=stimulus_id,
-                    )
-                ],
+                [LongRunningMsg(key=ts.key, compute_duration=compute_duration)],
             ),
             self._ensure_computing(),
         )

distributed/worker_state_machine.py

@@ -342,10 +342,9 @@ class RescheduleMsg(SendMessageToScheduler):
 class LongRunningMsg(SendMessageToScheduler):
     op = "long-running"
 
-    __slots__ = ("key", "compute_duration", "stimulus_id")
+    __slots__ = ("key", "compute_duration")
     key: str
     compute_duration: float
-    stimulus_id: str
 
 
 @dataclass