From 6354c90adeaa880ccb0b27bc239ff47413ae9ad5 Mon Sep 17 00:00:00 2001
From: Jiya Gupta <141018273+Jiya873@users.noreply.github.com>
Date: Mon, 15 Dec 2025 23:00:53 +0530
Subject: [PATCH 1/3] Refactor: rename AutoregressiveComponent to
 Autoregressive

---
 .../models/structural/components/autoregressive.py          | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/pymc_extras/statespace/models/structural/components/autoregressive.py b/pymc_extras/statespace/models/structural/components/autoregressive.py
index e20c5caa7..56dcc0551 100644
--- a/pymc_extras/statespace/models/structural/components/autoregressive.py
+++ b/pymc_extras/statespace/models/structural/components/autoregressive.py
@@ -6,7 +6,7 @@
 from pymc_extras.statespace.utils.constants import AR_PARAM_DIM
 
 
-class AutoregressiveComponent(Component):
+class Autoregressive(Component):
     r"""
     Autoregressive timeseries component
 
@@ -59,7 +59,7 @@ class AutoregressiveComponent(Component):
         import pytensor.tensor as pt
 
         trend = st.LevelTrendComponent(order=1, innovations_order=0)
-        ar = st.AutoregressiveComponent(2)
+        ar = st.Autoregressive(2)
         ss_mod = (trend + ar).build()
 
         with pm.Model(coords=ss_mod.coords) as model:
@@ -211,3 +211,5 @@ def make_symbolic_graph(self) -> None:
 
         cov_idx = ("state_cov", *np.diag_indices(k_posdef))
         self.ssm[cov_idx] = sigma_ar**2
+
+AutoregressiveComponent = Autoregressive

From 88f5b551b1359356835fbf3495d3a9e7206f309d Mon Sep 17 00:00:00 2001
From: Jiya Gupta <141018273+Jiya873@users.noreply.github.com>
Date: Mon, 15 Dec 2025 23:07:30 +0530
Subject: [PATCH 2/3] Refactor: rename AutoregressiveComponent to
 Autoregressive

---
 .../statespace/models/structural/components/autoregressive.py   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pymc_extras/statespace/models/structural/components/autoregressive.py b/pymc_extras/statespace/models/structural/components/autoregressive.py
index 56dcc0551..b46bf13f5 100644
--- a/pymc_extras/statespace/models/structural/components/autoregressive.py
+++ b/pymc_extras/statespace/models/structural/components/autoregressive.py
@@ -212,4 +212,4 @@ def make_symbolic_graph(self) -> None:
         cov_idx = ("state_cov", *np.diag_indices(k_posdef))
         self.ssm[cov_idx] = sigma_ar**2
 
-AutoregressiveComponent = Autoregressive
+AutoregressiveComponent = Autoregressive
\ No newline at end of file

From afa43e6f9f20f7046d9286ab487a08065846e837 Mon Sep 17 00:00:00 2001
From: Jiya Gupta <141018273+Jiya873@users.noreply.github.com>
Date: Mon, 15 Dec 2025 23:17:41 +0530
Subject: [PATCH 3/3] Refactor: rename AutoregressiveComponent to
 Autoregressive

---
 .../statespace/models/structural/components/autoregressive.py  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pymc_extras/statespace/models/structural/components/autoregressive.py b/pymc_extras/statespace/models/structural/components/autoregressive.py
index b46bf13f5..b5662d0de 100644
--- a/pymc_extras/statespace/models/structural/components/autoregressive.py
+++ b/pymc_extras/statespace/models/structural/components/autoregressive.py
@@ -212,4 +212,5 @@ def make_symbolic_graph(self) -> None:
         cov_idx = ("state_cov", *np.diag_indices(k_posdef))
         self.ssm[cov_idx] = sigma_ar**2
 
-AutoregressiveComponent = Autoregressive
\ No newline at end of file
+
+AutoregressiveComponent = Autoregressive