Skip to content

Commit cb59c80

Browse files
committed
updated SARMA and custom statespace notebooks to ensure they run after latest changes made to state space and updated SARIMAX sigma shape specification to be consistent with statespace structural
1 parent 2024134 commit cb59c80

File tree

3 files changed

+326
-389
lines changed

3 files changed

+326
-389
lines changed

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.

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)