type change

DiamondLightSource · Oct 16, 2024 · 53dc1ad · 53dc1ad
1 parent a053fd3
commit 53dc1ad
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/blueapi/worker/event.py b/src/blueapi/worker/event.py
@@ -8,7 +8,8 @@
 from blueapi.utils import BlueapiBaseModel
 
 # The RunEngine can return any of these three types as its state
-RawRunEngineState = type[PropertyMachine | ProxyString | str]
+# RawRunEngineState = type[PropertyMachine | ProxyString | str]
+RawRunEngineState = PropertyMachine | ProxyString | str
 
 
 # NOTE this is interim until refactor

diff --git a/src/blueapi/worker/task_worker.py b/src/blueapi/worker/task_worker.py
@@ -96,7 +96,8 @@ def __init__(
         self._tasks = {}
 
         assert ctx.run_engine.state is not None, "RunEngine state is not set"
-        self._state = WorkerState.from_bluesky_state(ctx.run_engine.state)
+        state: RawRunEngineState = str(ctx.run_engine.state)
+        self._state = WorkerState.from_bluesky_state(state)
         self._errors = []
         self._warnings = []
         self._task_channel = Queue(maxsize=1)