check_requirements.py

#!/usr/bin/env python3
"""
Check if all required packages are installed.
"""

import importlib
import sys


def check_package(package_name, import_name=None):
    """Check if a package is installed."""
    if import_name is None:
        import_name = package_name
    
    try:
        importlib.import_module(import_name)
        return True
    except ImportError:
        return False


def main():
    """Check all required packages."""
    print("Checking required packages for Adaptive CoT Framework...")
    print("=" * 60)
    
    # Core dependencies
    core_packages = [
        ("torch", "torch"),
        ("transformers", "transformers"),
        ("numpy", "numpy"),
        ("scipy", "scipy"),
        ("scikit-learn", "sklearn"),
        ("datasets", "datasets"),
        ("accelerate", "accelerate"),
        ("evaluate", "evaluate"),
        ("pyyaml", "yaml"),
        ("tqdm", "tqdm"),
        ("pandas", "pandas"),
        ("matplotlib", "matplotlib"),
        ("seaborn", "seaborn"),
    ]
    
    # Optional dependencies
    optional_packages = [
        ("vllm", "vllm"),
        ("lighteval", "lighteval"),
    ]
    
    print("Core Dependencies:")
    print("-" * 30)
    
    missing_core = []
    for package, import_name in core_packages:
        if check_package(package, import_name):
            print(f"✓ {package}")
        else:
            print(f"✗ {package} - MISSING")
            missing_core.append(package)
    
    print("\nOptional Dependencies:")
    print("-" * 30)
    
    missing_optional = []
    for package, import_name in optional_packages:
        if check_package(package, import_name):
            print(f"✓ {package}")
        else:
            print(f"✗ {package} - MISSING (optional)")
            missing_optional.append(package)
    
    print("\n" + "=" * 60)
    
    if missing_core:
        print(f"❌ Missing core dependencies: {', '.join(missing_core)}")
        print("\nInstall missing packages with:")
        print(f"pip install {' '.join(missing_core)}")
        return False
    else:
        print("✅ All core dependencies are installed!")
    
    if missing_optional:
        print(f"\n⚠️  Missing optional dependencies: {', '.join(missing_optional)}")
        print("These are optional but recommended for full functionality:")
        print(f"pip install {' '.join(missing_optional)}")
    
    print("\n🎉 Framework is ready to use!")
    return True


if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)