update param_dims

jessegrabowski · jessegrabowski · commit 8144ce297561 · 2025-08-05T10:59:17.000+08:00
diff --git a/pymc_extras/statespace/models/structural/components/cycle.py b/pymc_extras/statespace/models/structural/components/cycle.py
@@ -258,7 +258,7 @@ def populate_component_properties(self):
                 }
             }
         else:
-            self.param_dims = {self.name: (f"endog_{self.name}", f"state_{self.name}")}
+            self.param_dims = {f"params_{self.name}": (f"endog_{self.name}", f"state_{self.name}")}
             self.coords = {
                 f"state_{self.name}": [f"Cos_{self.name}", f"Sin_{self.name}"],
                 f"endog_{self.name}": self.observed_state_names,

Original file line number	Diff line number	Diff line change
`@@ -258,7 +258,7 @@ def populate_component_properties(self):`
`258`	`258`	`}`
`259`	`259`	`}`
`260`	`260`	`else:`
`261`		`- self.param_dims = {self.name: (f"endog_{self.name}", f"state_{self.name}")}`
	`261`	`+ self.param_dims = {f"params_{self.name}": (f"endog_{self.name}", f"state_{self.name}")}`
`262`	`262`	`self.coords = {`
`263`	`263`	`f"state_{self.name}": [f"Cos_{self.name}", f"Sin_{self.name}"],`
`264`	`264`	`f"endog_{self.name}": self.observed_state_names,`