add __all__ to models.__init__

ihmeuw-msca · Dec 26, 2024 · 78cf530 · 78cf530
1 parent 1ec6290
commit 78cf530
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/regmod/models/__init__.py b/src/regmod/models/__init__.py
@@ -10,3 +10,20 @@
 from .poisson import CanonicalPoissonModel, PoissonModel, create_poisson_model
 from .tobit import TobitModel
 from .weibull import WeibullModel
+
+__all__ = [
+    "BinomialModel",
+    "CanonicalBinomialModel",
+    "create_binomial_model",
+    "CanonicalGaussianModel",
+    "GaussianModel",
+    "create_gaussian_model",
+    "Model",
+    "NegativeBinomialModel",
+    "PogitModel",
+    "CanonicalPoissonModel",
+    "PoissonModel",
+    "create_poisson_model",
+    "TobitModel",
+    "WeibullModel",
+]