#10: State default inference partial completion.

src/upstage_des/states.py

@@ -226,14 +226,16 @@ def __set_name__(self, owner: "Actor", name: str) -> None:
     def _infer_state(self, instance: "Actor") -> tuple[Any, ...]:
         """Infer types for the state.
 
+        This should allow isinstance(value, self._infer_state(instance))
+
         Args:
             instance (Actor): The actor the state is attached to.
 
         Returns:
-            le[Any,...]: The state type
+            tuple[Any,...]: The state type
         """
         state_class = instance._state_defs[self.name]
-        args = get_args(state_class.__orig_class__)
+        args = get_args(state_class.__orig_class__) # type: ignore [attr-defined]
         return args
 
     def _set_default(self, instance: "Actor") -> None:

src/upstage_des/test/test_state.py

@@ -385,3 +385,19 @@ class Worker(UP.Actor):
         assert state_name is not None
         value = getattr(worker, state_name)
         assert value is worker.walkie, "Wrong state retrieved"
+
+
+def test_type_inference() -> None:
+    class A(UP.Actor):
+        st = UP.State[int | float]()
+
+    with UP.EnvironmentContext():
+        a = A(name="hi", st=1)
+        print(a.st)
+
+        v = a._state_defs["st"]._infer_state(a)
+        print(v)
+
+
+if __name__ == "__main__":
+    test_type_inference()