src/zen_translator/training/__init__.py

"""Training infrastructure for Zen Translator."""

from .news_anchor_dataset import (
    NEWS_CHANNELS,
    NewsAnchorDatasetBuilder,
    NewsAnchorSample,
    build_news_anchor_dataset,
)
from .swift_config import (
    NewsAnchorConfig,
    SwiftTrainingConfig,
    ZenIdentityConfig,
    create_training_dataset,
    generate_identity_dataset,
)

__all__ = [
    "SwiftTrainingConfig",
    "ZenIdentityConfig",
    "NewsAnchorConfig",
    "create_training_dataset",
    "generate_identity_dataset",
    "NewsAnchorDatasetBuilder",
    "NewsAnchorSample",
    "NEWS_CHANNELS",
    "build_news_anchor_dataset",
]