Skip to content

Add public API exports to link_predictor module - #252

Open
datatools-agent-2026 wants to merge 1 commit into
Intugle:mainfrom
datatools-agent-2026:cursor/data-ai-contribution-9a04
Open

Add public API exports to link_predictor module#252
datatools-agent-2026 wants to merge 1 commit into
Intugle:mainfrom
datatools-agent-2026:cursor/data-ai-contribution-9a04

Conversation

@datatools-agent-2026

Copy link
Copy Markdown

Summary

Adds public API exports to src/intugle/link_predictor/__init__.py so users can import the main link predictor classes directly from the package:

from intugle.link_predictor import LinkPredictor, PredictedLink, LinkPredictionResult

instead of verbose submodule imports.

Changes

  • Add module docstring describing the link predictor package
  • Export LinkPredictor, LinkPredictionSaver, and NoLinksFoundError from predictor
  • Export PredictedLink and LinkPredictionResult from models
  • Define __all__ for an explicit public API
  • Add regression tests in tests/link_predictor/test_public_api.py

Testing

  • python -c "from intugle.link_predictor import LinkPredictor; print('OK')" — passes
  • pytest tests/link_predictor/test_public_api.py — 2/2 passed
  • ruff check src/intugle/link_predictor/__init__.py — passes
  • pytest tests/link_predictor/ — 16/23 passed (7 integration tests require LLM/API credentials)

Fixes #127

Export LinkPredictor, LinkPredictionSaver, and NoLinksFoundError from
predictor; export PredictedLink and LinkPredictionResult from models.
Add module docstring and __all__ for explicit public API.

Add regression tests verifying package-level imports work correctly.

Fixes Intugle#127
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants