diff --git a/marginaleffects/__init__.py b/marginaleffects/__init__.py
index e69de29..c6ab264 100644
--- a/marginaleffects/__init__.py
+++ b/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
diff --git a/pyproject.toml b/pyproject.toml
index ef22c46..52bca9f 100644
--- a/pyproject.toml
+++ b/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"
diff --git a/tests/test_hypotheses.py b/tests/test_hypotheses.py
index ae38e30..2b6fb34 100644
--- a/tests/test_hypotheses.py
+++ b/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]))