diff --git a/src/snanosm/mealy.py b/src/snanosm/mealy.py index 6956b7a..0016671 100644 --- a/src/snanosm/mealy.py +++ b/src/snanosm/mealy.py @@ -112,7 +112,7 @@ class Machine(Generic[Input]): # dict[hash_origin, dict[hash_transition_input, Transition]] if self.__current_state_hash not in self.__transitions.keys(): raise ValueError("Processing input for a state with no transitions.") - # noinspection PyTypeChecker + assert self.__current_state_hash is not None transitions = self.__transitions[self.__current_state_hash] hi = hash(i) transition: Optional[Transition] = None