# Pin key packages for deterministic integration tests.
# These versions must match the local dev environment so that
# seeded loss values stay consistent across CI and local runs.
torch==2.11.0
torchvision==0.26.0
numpy==2.4.4
timm==1.0.26
transformers==5.4.0
lightning==2.6.1
# datasets>=2.20 is required for pyarrow>=21 (older datasets imported the removed
# pyarrow.PyExtensionType). Pinned-floor so CI's --user install upgrades any
# stale datasets; see pyproject for the full rationale.
datasets>=2.20.0
