Add BayesianETS and compile_statespace to statespace.__all__

jessegrabowski · jessegrabowski · commit d8f4863857e8 · 2024-07-29T23:48:33.000+08:00
diff --git a/pymc_experimental/statespace/__init__.py b/pymc_experimental/statespace/__init__.py
@@ -1,5 +1,7 @@
 from pymc_experimental.statespace.models import structural
+from pymc_experimental.statespace.models.ETS import BayesianETS
 from pymc_experimental.statespace.models.SARIMAX import BayesianSARIMA
 from pymc_experimental.statespace.models.VARMAX import BayesianVARMAX
+from pymc_experimental.statespace.utils import compile_statespace
 
-__all__ = ["structural", "BayesianSARIMA", "BayesianVARMAX"]
+__all__ = ["structural", "BayesianSARIMA", "BayesianVARMAX", "BayesianETS", "compile_statespace"]
diff --git a/pymc_experimental/statespace/models/__init__.py b/pymc_experimental/statespace/models/__init__.py
@@ -1,5 +1,6 @@
 from pymc_experimental.statespace.models import structural
+from pymc_experimental.statespace.models.ETS import BayesianETS
 from pymc_experimental.statespace.models.SARIMAX import BayesianSARIMA
 from pymc_experimental.statespace.models.VARMAX import BayesianVARMAX
 
-__all__ = ["structural", "BayesianSARIMA", "BayesianVARMAX"]
+__all__ = ["structural", "BayesianSARIMA", "BayesianVARMAX", "BayesianETS"]
diff --git a/pymc_experimental/statespace/utils/__init__.py b/pymc_experimental/statespace/utils/__init__.py
@@ -0,0 +1,3 @@
+from pymc_experimental.statespace.utils.compile import compile_statespace
+
+__all__ = ["compile_statespace"]

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+from pymc_experimental.statespace.utils.compile import compile_statespace`
	`2`	`+`
	`3`	`+__all__ = ["compile_statespace"]`