training/__init__.py

"""
Training module for Speech-to-Speech finetuning.

Follows SOLID principles:
- Single Responsibility: Each module handles one concern
- Open/Closed: Base classes allow extension without modification
- Dependency Inversion: Components depend on abstractions
"""

from .config import TrainingConfig, GPUConfig
from .data import load_sharded_dataset, InterleavedDataset, create_dataloader
from .models import SpeechAdapter
from .interleaving import create_interleaved_sequence, get_text_ratio
from .checkpoint import CheckpointManager
from .utils import setup_logging, log, get_device_info

__all__ = [
    'TrainingConfig',
    'GPUConfig',
    'load_sharded_dataset',
    'InterleavedDataset',
    'create_dataloader',
    'SpeechAdapter',
    'create_interleaved_sequence',
    'get_text_ratio',
    'CheckpointManager',
    'setup_logging',
    'log',
    'get_device_info',
]