@@ -98,8 +98,7 @@ class StateMachineDevice(DeviceBase, CanProcessComposite):
9898 def __init__ (
9999 self ,
100100 override_states : dict [str , State ] | None = None ,
101- override_transitions : dict [tuple [State , State ], Callable [[], bool ]]
102- | None = None ,
101+ override_transitions : dict [tuple [State , State ], Callable [[], bool ]] | None = None ,
103102 override_initial_state : State | None = None ,
104103 override_initial_data : dict [str , float ] | None = None ,
105104 ) -> None :
@@ -115,17 +114,13 @@ def __init__(
115114 initial = override_initial_state or self ._get_initial_state ()
116115
117116 if initial not in state_handlers :
118- raise RuntimeError (
119- "Initial state '{}' is not a valid state." .format (initial )
120- )
117+ raise RuntimeError ("Initial state '{}' is not a valid state." .format (initial ))
121118
122119 self ._csm = StateMachine (
123120 {
124121 "initial" : initial ,
125122 "states" : state_handlers ,
126- "transitions" : self ._get_final_transition_handlers (
127- override_transitions
128- ),
123+ "transitions" : self ._get_final_transition_handlers (override_transitions ),
129124 },
130125 context = self ,
131126 )
@@ -151,9 +146,7 @@ def _get_initial_state(self) -> str:
151146
152147 :return: The initial state of the state machine.
153148 """
154- raise NotImplementedError (
155- "_get_initial_state must be implemented in a StateMachineDevice."
156- )
149+ raise NotImplementedError ("_get_initial_state must be implemented in a StateMachineDevice." )
157150
158151 def _get_transition_handlers (self ) -> dict [tuple [str , str ], Callable [[], bool ]]:
159152 """
@@ -176,9 +169,7 @@ def _initialize_data(self) -> None:
176169 """
177170 pass
178171
179- def _get_final_state_handlers (
180- self , overrides : dict [str , State ] | None
181- ) -> dict [str , State ]:
172+ def _get_final_state_handlers (self , overrides : dict [str , State ] | None ) -> dict [str , State ]:
182173 states = self ._get_state_handlers ()
183174
184175 if overrides is not None :
0 commit comments