# Official fairseq-signals Requirements - ECG-FM
# EXACT requirements from Jwoo5/fairseq_signals setup.py
# NOTE: NumPy is installed separately in Dockerfile (1.26.4) for dependency compatibility

# Core FastAPI dependencies
fastapi==0.104.1
uvicorn[standard]==0.24.0
huggingface-hub==0.19.4

# CRITICAL FIX: Upgrade PyTorch to 2.1.0 for ECG-FM compatibility
# ECG-FM checkpoints require PyTorch >=2.1.0 for torch.nn.utils.parametrizations.weight_norm
# PyTorch 1.13.1 is missing this function, causing model loading failures
torch==2.1.0
torchvision==0.16.0
torchaudio==2.1.0

# CRITICAL FIX: Pin transformers to compatible version
# fairseq_signals requires transformers>=4.21.0 but transformers 4.55.4 has breaking changes
# transformers 4.21.0 is the last version with GenerationMixin in transformers.generation
transformers==4.21.0

# CRITICAL FIX: Pin omegaconf to compatible version
# ECG-FM checkpoints require omegaconf <2.4 that has is_primitive_type function
# omegaconf 2.1.2 is the last version with this function
omegaconf==2.1.2

# System monitoring for status endpoint
psutil>=5.9.0

# Data processing for ECG segmentation and aggregation
pandas>=1.3.0

# Official fairseq-signals dependencies (EXACT from their setup.py)
# NOTE: NumPy is managed separately in Dockerfile to avoid version conflicts
cffi
cython
hydra-core
regex
sacrebleu>=1.4.12
tqdm
scikit-learn

# Additional dependencies for building
setuptools>=65.0.0
wheel>=0.38.0