Fix AR labeling + core docstrings typo

AlexAndorra · AlexAndorra · commit 5590d67317a5 · 2025-07-21T11:07:14.000-04:00
diff --git a/pymc_extras/statespace/models/structural/components/autoregressive.py b/pymc_extras/statespace/models/structural/components/autoregressive.py
@@ -114,7 +114,7 @@ def populate_component_properties(self):
         if self.k_endog > 1:
             self.param_dims[f"params_{self.name}"] = (
                 f"endog_{self.name}",
-                AR_PARAM_DIM,
+                f"lag_{self.name}",
             )
             self.param_dims[f"sigma_{self.name}"] = (f"endog_{self.name}",)
 
diff --git a/pymc_extras/statespace/models/structural/core.py b/pymc_extras/statespace/models/structural/core.py
@@ -362,7 +362,7 @@ def extract_components_from_idata(self, idata: xr.Dataset) -> xr.Dataset:
         Returns
         -------
         idata: Dataset
-            An Dataset object with hidden states transformed to represent only the "interpretable" subcomponents
+            A Dataset object with hidden states transformed to represent only the "interpretable" subcomponents
             of the structural model.
 
         Notes

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ def populate_component_properties(self):`
`114`	`114`	`if self.k_endog > 1:`
`115`	`115`	`self.param_dims[f"params_{self.name}"] = (`
`116`	`116`	`f"endog_{self.name}",`
`117`		`- AR_PARAM_DIM,`
	`117`	`+ f"lag_{self.name}",`
`118`	`118`	`)`
`119`	`119`	`self.param_dims[f"sigma_{self.name}"] = (f"endog_{self.name}",)`
`120`	`120`