File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 66from pymc_extras .statespace .utils .constants import AR_PARAM_DIM
77
88
9- class AutoregressiveComponent (Component ):
9+ class Autoregressive (Component ):
1010 r"""
1111 Autoregressive timeseries component
1212
@@ -59,7 +59,7 @@ class AutoregressiveComponent(Component):
5959 import pytensor.tensor as pt
6060
6161 trend = st.LevelTrendComponent(order=1, innovations_order=0)
62- ar = st.AutoregressiveComponent (2)
62+ ar = st.Autoregressive (2)
6363 ss_mod = (trend + ar).build()
6464
6565 with pm.Model(coords=ss_mod.coords) as model:
@@ -211,3 +211,5 @@ def make_symbolic_graph(self) -> None:
211211
212212 cov_idx = ("state_cov" , * np .diag_indices (k_posdef ))
213213 self .ssm [cov_idx ] = sigma_ar ** 2
214+
215+ AutoregressiveComponent = Autoregressive
You can’t perform that action at this time.
0 commit comments