test_env.py

#!/usr/bin/env python3
"""Environment test script for HuggingFace Space"""

import sys
import torch

print("="*60)
print("ENVIRONMENT TEST")
print("="*60)

# Python version
print(f"Python: {sys.version}")

# CUDA check
print(f"\nCUDA available: {torch.cuda.is_available()}")
if torch.cuda.is_available():
    print(f"GPU: {torch.cuda.get_device_name(0)}")
    print(f"VRAM: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")

# Chronos 2 import
try:
    from chronos import Chronos2Pipeline
    import chronos
    print(f"\nChronos version: {chronos.__version__}")
    print("✓ Chronos 2 imported successfully")
except ImportError as e:
    print(f"\n✗ Chronos 2 import failed: {e}")

print("\n" + "="*60)
print("Test complete!")