# ProAffinity-GNN Requirements
# Corrected version with NumPy compatibility fix

# Core dependencies - INSTALL THESE FIRST
# Use the PyTorch index for CUDA-enabled versions:
# pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121

torch==2.2.2
torchvision==0.17.2
torchaudio==2.2.2

# NumPy - MUST be < 2.0 for PyTorch 2.2.2 compatibility
numpy<2.0  # Compatible version: 1.26.4

# Graph neural network
torch_geometric==2.3.0

# Transformers and NLP
transformers==4.38
tokenizers==0.15.2

# Scientific computing
scikit-learn>=1.5.0
scipy>=1.10.1

# Utilities
requests>=2.32.4
tqdm>=4.67.1
PyYAML>=6.0.2
psutil>=7.0.0
packaging>=25.0
filelock>=3.13.1
huggingface-hub>=0.33.2
safetensors>=0.5.3
regex>=2024.11.6
Jinja2>=3.1.6
joblib>=1.4.2
threadpoolctl>=3.5.0
pillow>=10.3.0

# Optional but recommended
certifi>=2025.4.26
charset-normalizer>=3.4.2
idna>=3.10
urllib3>=2.5.0
MarkupSafe>=2.1.5
mpmath>=1.3.0
networkx>=3.2.1
sympy>=1.13.3
typing_extensions>=4.12.2
pyparsing>=3.1.4
fsspec>=2024.6.1
hf-xet>=1.1.2

# Note: CUDA dependencies (nvidia-*) will be automatically installed with torch+cu121
