Skip to content

Commit 1ae433f

Browse files
committed
removed data & coords setters in _set<foo> medthod in Component class and placed default shoch and state setters
1 parent 228acff commit 1ae433f

File tree

1 file changed

+11
-8
lines changed
  • pymc_extras/statespace/models/structural

1 file changed

+11
-8
lines changed

pymc_extras/statespace/models/structural/core.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -683,17 +683,20 @@ def make_and_register_data(self, name, shape, dtype=floatX) -> Variable:
683683
def _set_parameters(self) -> None:
684684
raise NotImplementedError
685685

686-
def _set_data(self) -> None:
687-
raise NotImplementedError
688-
689686
def _set_shocks(self) -> None:
690-
raise NotImplementedError
687+
return ShockInfo(shocks=[Shock(name=f"shock_{n}") for n in range(self.k_posdef or 0)])
691688

692689
def _set_states(self) -> None:
693-
raise NotImplementedError
694-
695-
def _set_coords(self) -> None:
696-
raise NotImplementedError
690+
return StateInfo(
691+
states=[
692+
State(name=n, observed=False, shared=self.share_states)
693+
for n in range(self.k_states or 0)
694+
]
695+
+ [
696+
State(name=n, observed=True, shared=self.share_states)
697+
for n in range(self.k_endog or 0)
698+
]
699+
)
697700

698701
def make_symbolic_graph(self) -> None:
699702
raise NotImplementedError

0 commit comments

Comments
 (0)