Skip to content

Commit 1e87fd7

Browse files
committed
ruff using our config
1 parent e123ee8 commit 1e87fd7

File tree

1 file changed

+8
-16
lines changed

1 file changed

+8
-16
lines changed

lewis/devices/__init__.py

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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[State, State], 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:
@@ -207,8 +198,9 @@ def _override_data(self, overrides: dict[str, float] | None) -> None:
207198
self.log.debug("Trying to override initial data (%s=%s)", name, val)
208199
if name not in dir(self):
209200
raise AttributeError(
210-
"Can not override non-existing attribute"
211-
"'{}' of class '{}'.".format(name, type(self).__name__)
201+
"Can not override non-existing attribute'{}' of class '{}'.".format(
202+
name, type(self).__name__
203+
)
212204
)
213205

214206
setattr(self, name, val)

0 commit comments

Comments
 (0)