Skip to content

Commit 6354c90

Browse files
committed
Refactor: rename AutoregressiveComponent to Autoregressive
1 parent 3531d29 commit 6354c90

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from 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

0 commit comments

Comments
 (0)