Skip to content

Conversation

jessegrabowski
Copy link
Member

Exogenous regressors are added via a time-varying observation bias to keep the statespace as small as possible.

We could consider allowing something fancier, but this is a good v0.

@jessegrabowski jessegrabowski added enhancements New feature or request statespace labels Aug 21, 2025
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Copy link

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR renames BayesianSARIMA to BayesianSARIMAX and adds support for exogenous regressors in the model. The X suffix follows the standard convention for models with exogenous variables.

Key changes:

  • Renamed the class from BayesianSARIMA to BayesianSARIMAX
  • Added support for exogenous regressors via new parameters k_exog and exog_state_names
  • Implemented exogenous data handling through observation intercept modifications

Reviewed Changes

Copilot reviewed 8 out of 9 changed files in this pull request and generated 3 comments.

Show a summary per file
File Description
pymc_extras/statespace/models/SARIMAX.py Main model class renamed and exogenous regression functionality added
tests/statespace/models/test_SARIMAX.py Updated tests to use new class name and added exogenous regression test
pymc_extras/statespace/core/statespace.py Updated data handling for exogenous variables
pymc_extras/statespace/models/__init__.py Updated imports and exports for renamed class
pymc_extras/statespace/__init__.py Updated public API exports
Various other files Updated references and documentation

Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.

Copy link
Contributor

@Dekermanjian Dekermanjian left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Jesse this looks great!

@jessegrabowski jessegrabowski changed the title Allow exogenous regressions in BayiesnaSARIMAX Allow exogenous regressions in BayesianSARIMAX Aug 24, 2025
@jessegrabowski jessegrabowski merged commit 89c6bc0 into main Aug 24, 2025
28 of 30 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancements New feature or request statespace
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants