Merge branch 'main' into story_tags

distributed/worker_state_machine.py

@@ -2774,10 +2774,14 @@ def _handle_refresh_who_has(self, ev: RefreshWhoHasEvent) -> RecsInstrs:
     # Diagnostics #
     ###############
 
-    def story(self, *keys_or_tasks: str | TaskState) -> list[tuple]:
-        """Return all transitions involving one or more tasks"""
-        keys = {e.key if isinstance(e, TaskState) else e for e in keys_or_tasks}
-        return worker_story(keys, self.log)
+    def story(self, *keys_or_tasks_or_stimuli: str | TaskState) -> list[tuple]:
+        """Return all records from the transitions log involving one or more tasks or
+        stimulus_id's
+        """
+        keys_or_stimuli = {
+            e.key if isinstance(e, TaskState) else e for e in keys_or_tasks_or_stimuli
+        }
+        return worker_story(keys_or_stimuli, self.log)
 
     def stimulus_story(
         self, *keys_or_tasks: str | TaskState