Skip to content

Commit 6d6c74e

Browse files
committed
Merge branch 'main' into better-extract-component
2 parents 4152d4a + 2d16ad0 commit 6d6c74e

File tree

5 files changed

+838
-1336
lines changed

5 files changed

+838
-1336
lines changed

notebooks/Exponential Trend Smoothing.ipynb

Lines changed: 413 additions & 840 deletions
Large diffs are not rendered by default.

notebooks/Making a Custom Statespace Model.ipynb

Lines changed: 109 additions & 47 deletions
Large diffs are not rendered by default.

notebooks/SARMA Example.ipynb

Lines changed: 215 additions & 340 deletions
Large diffs are not rendered by default.

notebooks/VARMAX Example.ipynb

Lines changed: 99 additions & 107 deletions
Large diffs are not rendered by default.

pymc_extras/statespace/models/SARIMAX.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -293,11 +293,11 @@ def param_info(self) -> dict[str, dict[str, Any]]:
293293
"constraints": "Positive Semi-definite",
294294
},
295295
"sigma_obs": {
296-
"shape": None if self.k_endog == 1 else (self.k_endog,),
296+
"shape": () if self.k_endog == 1 else (self.k_endog,),
297297
"constraints": "Positive",
298298
},
299299
"sigma_state": {
300-
"shape": None if self.k_posdef == 1 else (self.k_posdef,),
300+
"shape": () if self.k_posdef == 1 else (self.k_posdef,),
301301
"constraints": "Positive",
302302
},
303303
"ar_params": {

0 commit comments

Comments
 (0)