# Core ML dependencies
torch>=1.13.0
torchvision>=0.14.0
torchaudio>=0.13.0
numpy>=1.21.0,<2.0.0

# Configuration management
hydra-core>=1.2.0
omegaconf>=2.2.0

# Web framework for deployment
fastapi>=0.68.0
uvicorn>=0.15.0
websockets>=10.0
python-multipart>=0.0.5
requests>=2.25.0

# Image processing
opencv-python-headless>=4.5.0
Pillow>=8.0.0
pygame>=2.1.0

# Hugging Face integration
huggingface-hub>=0.10.0

# Data handling
h5py>=3.7.0

# Atari environment dependencies
ale_py>=0.8.0
gymnasium>=0.28.0

# Experiment tracking (optional, for training only)
# wandb>=0.13.0  # Commented out due to NumPy 2.0 compatibility issues

# Metrics (required by rew_end_model.py)
torcheval>=0.0.6

# Progress bars (may be used by various components)
tqdm>=4.64.0

# Optional: for better performance
# torch-audio  # if needed for audio processing

# Development dependencies (uncomment for local development)
# pytest>=6.0.0
# black>=21.0.0
# isort>=5.0.0