pymc-devs · AlexAndorra · Jul 31, 2025 · Jul 21, 2025 · Jul 27, 2025 · Jul 31, 2025
diff --git a/notebooks/Exponential Trend Smoothing.ipynb b/notebooks/Exponential Trend Smoothing.ipynb
diff --git a/notebooks/Making a Custom Statespace Model.ipynb b/notebooks/Making a Custom Statespace Model.ipynb
diff --git a/notebooks/SARMA Example.ipynb b/notebooks/SARMA Example.ipynb
diff --git a/notebooks/VARMAX Example.ipynb b/notebooks/VARMAX Example.ipynb
diff --git a/pymc_extras/statespace/models/SARIMAX.py b/pymc_extras/statespace/models/SARIMAX.py
@@ -293,11 +293,11 @@ def param_info(self) -> dict[str, dict[str, Any]]:
                 "constraints": "Positive Semi-definite",
             },
             "sigma_obs": {
-                "shape": None if self.k_endog == 1 else (self.k_endog,),
+                "shape": () if self.k_endog == 1 else (self.k_endog,),
                 "constraints": "Positive",
             },
             "sigma_state": {
-                "shape": None if self.k_posdef == 1 else (self.k_posdef,),
+                "shape": () if self.k_posdef == 1 else (self.k_posdef,),
                 "constraints": "Positive",
             },
             "ar_params": {