slightly improve typing

pytransitions · May 7, 2024 · 5e03bc2 · 5e03bc2
1 parent 99d69d2
commit 5e03bc2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/Changelog.md b/Changelog.md
@@ -9,6 +9,9 @@
 - '_anchor' suffix has been removed for (py)graphviz cluster node anchors
 - local testing switched from [tox](https://github.com/tox-dev/tox) to [nox](https://github.com/wntrblm/nox)
 - PR #633: Remove surrounding whitespace from docstrings (thanks @artofhuman)
+- Typing:
+  + Made machine property mandatory in (Nested)EventData
+  + Transition.dest may be None
 
 ## 0.9.0 (September 2022)
 

diff --git a/transitions/core.pyi b/transitions/core.pyi
@@ -51,7 +51,7 @@ class Transition:
     dynamic_methods: List[str]
     condition_cls: Type[Condition]
     source: str
-    dest: str
+    dest: Optional[str]
     prepare: CallbackList
     before: CallbackList
     after: CallbackList
@@ -70,7 +70,7 @@ TransitionConfig = Union[Sequence[Union[str, Any]], Dict[str, Any], Transition]
 class EventData:
     state: Optional[State]
     event: Optional[Event]
-    machine: Optional[Machine]
+    machine: Machine
     model: object
     args: Iterable[Any]
     kwargs: Dict[str, Any]

diff --git a/transitions/extensions/nesting.pyi b/transitions/extensions/nesting.pyi
@@ -23,7 +23,7 @@ class NestedEvent(Event):
 class NestedEventData(EventData):
     state: Optional[NestedState]
     event: Optional[NestedEvent]
-    machine: Optional[HierarchicalMachine]
+    machine: HierarchicalMachine
     transition: Optional[NestedTransition]
     source_name: Optional[str]
     source_path: Optional[List[str]]