Pass share_states to super calls

Author: AlexAndorra

pymc_extras/statespace/models/structural/components/autoregressive.py

@@ -103,6 +103,7 @@ def __init__(
             combine_hidden_states=True,
             observed_state_names=observed_state_names,
             obs_state_idxs=np.tile(np.r_[[1.0], np.zeros(k_states - 1)], k_endog_effective),
+            share_states=share_states,
         )
 
     def populate_component_properties(self):

pymc_extras/statespace/models/structural/components/cycle.py

@@ -197,6 +197,7 @@ def __init__(
             combine_hidden_states=True,
             obs_state_idxs=obs_state_idx,
             observed_state_names=observed_state_names,
+            share_states=share_states,
         )
 
     def make_symbolic_graph(self) -> None:

pymc_extras/statespace/models/structural/components/level_trend.py

@@ -170,6 +170,7 @@ def __init__(
             obs_state_idxs=np.tile(
                 np.array([1.0] + [0.0] * (k_states - 1)), k_endog if not share_states else 1
             ),
+            share_states=share_states,
         )
 
     def populate_component_properties(self):

pymc_extras/statespace/models/structural/components/measurement_error.py

@@ -119,6 +119,7 @@ def __init__(
             measurement_error=True,
             combine_hidden_states=False,
             observed_state_names=observed_state_names,
+            share_states=share_states,
         )
 
     def populate_component_properties(self):

pymc_extras/statespace/models/structural/components/regression.py

@@ -132,6 +132,7 @@ def __init__(
             k_states=k_states * k_endog if not share_states else k_states,
             k_posdef=k_posdef * k_endog if not share_states else k_posdef,
             state_names=self.state_names,
+            share_states=share_states,
             observed_state_names=observed_state_names,
             measurement_error=False,
             combine_hidden_states=False,

pymc_extras/statespace/models/structural/components/seasonality.py

@@ -296,6 +296,7 @@ def __init__(
             obs_state_idxs=np.tile(
                 np.array([1.0] + [0.0] * (k_states - 1)), 1 if share_states else k_endog
             ),
+            share_states=share_states,
         )
 
     def populate_component_properties(self):
@@ -525,6 +526,7 @@ def __init__(
             k_posdef=k_states * int(self.innovations)
             if share_states
             else k_states * int(self.innovations) * k_endog,
+            share_states=share_states,
             observed_state_names=observed_state_names,
             measurement_error=False,
             combine_hidden_states=True,