Correct type hint for _get_initial_state()

ISISComputingGroup · Jan 13, 2025 · 13ce61b · 13ce61b
1 parent 39a618e
commit 13ce61b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lewis/devices/__init__.py b/lewis/devices/__init__.py
@@ -139,7 +139,7 @@ def _get_state_handlers(self) -> dict[str, State]:
             "_get_state_handlers must be implemented in a StateMachineDevice."
         )
 
-    def _get_initial_state(self) -> State:
+    def _get_initial_state(self) -> str:
         """
         Implement this method to return the initial state of the internal state machine.
         The default implementation raises a ``NotImplementedError``.