init

marginaleffects/__init__.py

@@ -0,0 +1,8 @@
+from .comparisons import avg_comparisons, comparisons
+from .datagrid import datagrid, datagridcf
+from .hypotheses import hypotheses
+from .plot_comparisons import plot_comparisons
+from .plot_predictions import plot_predictions
+from .plot_slopes import plot_slopes
+from .predictions import avg_predictions, predictions
+from .slopes import avg_slopes, slopes

pyproject.toml

@@ -28,6 +28,9 @@ pytest-xdist = "^3.3.1"
 bandit = "^1.7.5"
 ruff = ">0.0.283"
 
+[tool.ruff]
+ignore-init-module-imports = true
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

tests/test_hypotheses.py

@@ -11,8 +11,6 @@
 
 mod = smf.ols("Literacy ~ Pop1831 * Desertion", dat).fit()
 
-avg_predictions(mod, by="Region")
-
 
 def test_coefs():
     hyp_py = hypotheses(mod, hypothesis=np.array([1, -1, 0, 0]))