-e nemo-gym[dev] @ ../../
# CVE remediation: torch 2.5.1 → 2.12.0.
# torch 2.12+ defaults to CUDA 13.0 wheels on PyPI. For CUDA 12.x
# environments, use a cu12-specific wheel index via UV_EXTRA_INDEX_URL.
setuptools>=70
torch==2.12.0
torchvision==0.27.0
sacrebleu>=2.4
sentencepiece
unbabel-comet>=2.2
# GlotLID scoring dependencies.
numpy
fasttext-wheel==0.9.2
huggingface-hub
# scipy is a transitive dep of unbabel-comet (via comet.models.metrics →
# scipy.stats) but uv's resolution doesn't pull it automatically here —
# explicit pin avoids ModuleNotFoundError in the COMET Ray task.
scipy
# Compact Language Detector v2 (Google CLD2) Python binding. Pure C++
# extension with no Python deps. Prebuilt wheels exist through cp313, so on
# the repo-pinned 3.13.14 it installs from a wheel. On 3.14 there is
# no wheel yet but it could be built from CLD2 source (needs a C++ compiler).
pycld2>=0.42
