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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[GOOD FIRST ISSUE] Add public API exports to link_predictor __init__.py

2 participants