diff --git a/tests/__init__.py b/tests/__init__.py
index e69de29..8c9bd63 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -0,0 +1,24 @@
+"""
+Link Predictor Module
+
+This module provides functionality for predicting relationships between datasets
+based on column profiling, data type analysis, and LLM-based inference.
+"""
+
+from intugle.link_predictor.models import (
+    LinkPredictionResult,
+    PredictedLink,
+)
+from intugle.link_predictor.predictor import (
+    LinkPredictor,
+    LinkPredictionSaver,
+    NoLinksFoundError,
+)
+
+__all__ = [
+    "LinkPredictor",
+    "LinkPredictionSaver",
+    "PredictedLink",
+    "LinkPredictionResult",
+    "NoLinksFoundError",
+]