# sqry-nl Training Pipeline Requirements
# Python 3.10+ required
#
# NOTE: Versions are pinned for reproducibility (H6 mitigation).
# Update versions carefully with full test suite verification.
#
# Version compatibility (May 2026):
# - transformers 5.x includes the Trainer security fix and requires
#   tokenizers 0.22.x.
# - optimum-onnx still requires transformers<4.58.0, so export_onnx.py uses
#   torch.onnx plus onnxruntime quantization directly.
# - numpy 2.4.x is required for Python 3.13 wheels.
# - Updated 2026-05-31: torch 2.12.0, transformers 5.0.0, datasets 4.4.1

# Core ML Framework
torch==2.12.0
transformers==5.0.0
datasets==4.4.1

# ONNX Export and Quantization
onnx==1.21.0
onnxruntime==1.23.2

# Tokenizer (transformers 5.0.x requires <=0.23.0,>=0.22.0)
tokenizers==0.22.2

# Training acceleration
accelerate>=1.0.0

# Data Processing
numpy==2.4.6
pandas==2.3.3
scikit-learn==1.7.2

# Calibration
netcal==1.3.5

# Utilities
tqdm==4.67.3
pyyaml==6.0.3
typer==0.25.1
click==8.4.1
rich==15.0.0

# Testing (optional)
pytest==9.0.3
pytest-cov==7.0.0
