Add observed_states helper to StateInfo

Jesse Grabowski · Jesse Grabowski · commit c6a48fcbb62b · 2025-11-07T11:46:05.000-06:00
diff --git a/pymc_extras/statespace/core/properties.py b/pymc_extras/statespace/core/properties.py
@@ -167,6 +167,10 @@ def __str__(self) -> str:
             f"states: {[s.name for s in self.items]}\nobserved: {[s.observed for s in self.items]}"
         )
 
+    @property
+    def observed_states(self) -> tuple[State, ...]:
+        return tuple(s for s in self.items if s.observed)
+
 
 @dataclass(frozen=True)
 class Shock(Property):

Original file line number	Diff line number	Diff line change
`@@ -167,6 +167,10 @@ def __str__(self) -> str:`
`167`	`167`	`f"states: {[s.name for s in self.items]}\nobserved: {[s.observed for s in self.items]}"`
`168`	`168`	`)`
`169`	`169`
	`170`	`+ @property`
	`171`	`+ def observed_states(self) -> tuple[State, ...]:`
	`172`	`+ return tuple(s for s in self.items if s.observed)`
	`173`	`+`
`170`	`174`
`171`	`175`	`@dataclass(frozen=True)`
`172`	`176`	`class Shock(Property):`