#!/usr/bin/env python3
"""
Simple training runner script for AI Trainer
"""

import os
import sys
from pathlib import Path

# Add src to path
sys.path.append(str(Path(__file__).parent / "src"))

from main import main

if __name__ == "__main__":
    # Set environment variables for better CUDA performance
    os.environ['CUDA_VISIBLE_DEVICES'] = '0'
    os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:512'
    os.environ['TOKENIZERS_PARALLELISM'] = 'false'

    # Run the main training application
    main()