"""Quick test to verify imports work."""

import sys
import os

try:
    from core.formula import ScoringFormula
    print("[OK] Formula import successful")
except Exception as e:
    print(f"[FAIL] Formula import failed: {e}")
    sys.exit(1)

try:
    from model.architecture import NovelAIModel
    print("[OK] Model import successful")
except Exception as e:
    print(f"[FAIL] Model import failed: {e}")
    sys.exit(1)

try:
    from config.config import Config
    print("[OK] Config import successful")
except Exception as e:
    print(f"[FAIL] Config import failed: {e}")
    sys.exit(1)

try:
    from training.trainer import Trainer
    print("[OK] Trainer import successful")
except Exception as e:
    print(f"[FAIL] Trainer import failed: {e}")
    sys.exit(1)

print("\nAll imports successful!")