ruff

Author: rerpha

lewis/devices/__init__.py

@@ -98,7 +98,8 @@ class StateMachineDevice(DeviceBase, CanProcessComposite):
     def __init__(
         self,
         override_states: dict[str, State] | None = None,
-        override_transitions: dict[tuple[State, State], Callable[[], bool]] | None = None,
+        override_transitions: dict[tuple[State, State], Callable[[], bool]]
+        | None = None,
         override_initial_state: State | None = None,
         override_initial_data: dict[str, float] | None = None,
     ) -> None:
@@ -114,13 +115,17 @@ def __init__(
         initial = override_initial_state or self._get_initial_state()
 
         if initial not in state_handlers:
-            raise RuntimeError("Initial state '{}' is not a valid state.".format(initial))
+            raise RuntimeError(
+                "Initial state '{}' is not a valid state.".format(initial)
+            )
 
         self._csm = StateMachine(
             {
                 "initial": initial,
                 "states": state_handlers,
-                "transitions": self._get_final_transition_handlers(override_transitions),
+                "transitions": self._get_final_transition_handlers(
+                    override_transitions
+                ),
             },
             context=self,
         )
@@ -146,7 +151,9 @@ def _get_initial_state(self) -> str:
 
         :return: The initial state of the state machine.
         """
-        raise NotImplementedError("_get_initial_state must be implemented in a StateMachineDevice.")
+        raise NotImplementedError(
+            "_get_initial_state must be implemented in a StateMachineDevice."
+        )
 
     def _get_transition_handlers(self) -> dict[tuple[State, State], Callable[[], bool]]:
         """
@@ -169,7 +176,9 @@ def _initialize_data(self) -> None:
         """
         pass
 
-    def _get_final_state_handlers(self, overrides: dict[str, State] | None) -> dict[str, State]:
+    def _get_final_state_handlers(
+        self, overrides: dict[str, State] | None
+    ) -> dict[str, State]:
         states = self._get_state_handlers()
 
         if overrides is not None:
@@ -198,9 +207,8 @@ def _override_data(self, overrides: dict[str, float] | None) -> None:
                 self.log.debug("Trying to override initial data (%s=%s)", name, val)
                 if name not in dir(self):
                     raise AttributeError(
-                        "Can not override non-existing attribute" "'{}' of class '{}'.".format(
-                            name, type(self).__name__
-                        )
+                        "Can not override non-existing attribute"
+                        "'{}' of class '{}'.".format(name, type(self).__name__)
                     )
 
                 setattr(self, name, val)