Add descriptive error message to assert

dask · Aug 11, 2022 · 7734557 · 7734557
1 parent cb45c61
commit 7734557
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/distributed/worker_state_machine.py b/distributed/worker_state_machine.py
@@ -1918,7 +1918,10 @@ def _transition_cancelled_error(
         *,
         stimulus_id: str,
     ) -> RecsInstrs:
-        assert ts.previous in ("executing", "long-running")
+        assert ts.previous in (
+            "executing",
+            "long-running",
+        ), f"Expected 'executing' or 'long-running'; got '{ts.previous}'"
         recs, instructions = self._transition_executing_error(
             ts,
             exception,