"""
Training CLI wrapper for tumor detection package.

This module provides the command-line interface for training models.
"""

import sys
from pathlib import Path


def main():
    """Main entry point for training CLI."""
    # Add the project root to sys.path to access existing training scripts
    project_root = Path(__file__).parent.parent.parent.parent
    sys.path.insert(0, str(project_root))

    # Import and run the existing training script
    try:
        from src.training.train_enhanced import main as train_main
        train_main()
    except ImportError:
        # Fallback to basic training script
        from src.training.train import main as train_main
        train_main()


if __name__ == "__main__":
    main()
