| 1 | -r ../common.txt |
| 2 | |
| 3 | # testing |
| 4 | pytest |
| 5 | tensorizer==2.10.1 |
| 6 | pytest-forked |
| 7 | pytest-asyncio |
| 8 | pytest-rerunfailures |
| 9 | pytest-shard |
| 10 | pytest-timeout |
| 11 | pytest-cov |
| 12 | |
| 13 | # testing utils |
| 14 | albumentations # required for Nemotron Parse in test_common.py |
| 15 | av # required for audio_in_video tests |
| 16 | backoff # required for phi4mm test |
| 17 | blobfile # required for kimi-vl test |
| 18 | httpx |
| 19 | librosa # required for audio tests |
| 20 | vector_quantize_pytorch # required for minicpmo_26 test |
| 21 | vocos # required for minicpmo_26 test |
| 22 | peft>=0.19.1 # required for phi-4-mm test |
| 23 | pqdm |
| 24 | ray[cgraph,default]==2.55.1 # Includes worker startup getenv/setenv race fix |
| 25 | sentence-transformers>=5.2.0 # required for embedding tests |
| 26 | soundfile # required for audio tests |
| 27 | jiwer # required for audio tests |
| 28 | tblib # for pickling test exceptions |
| 29 | timm>=1.0.17 # required for internvl and gemma3n-mm test |
| 30 | transformers_stream_generator # required for qwen-vl test |
| 31 | matplotlib # required for qwen-vl test |
| 32 | mistral_common[image,audio]>=1.11.6 # required for voxtral test |
| 33 | num2words # required for smolvlm test |
| 34 | open_clip_torch==2.32.0 # Required for nemotron_vl test, Nemotron Parse in test_common.py |
| 35 | datamodel_code_generator>=0.64.0 # required for minicpm3 test |
| 36 | nltk>=3.10.3 # pin transitive rouge-score/lm-eval dep for known GHSAs |
| 37 | lm-eval[api]>=0.4.12 # required for model evaluation test |
| 38 | mteb[bm25s]>=2, <3 # required for mteb test |
| 39 | transformers==5.17.0 |
| 40 | schemathesis>=4.0.0 # Required for openai schema test |
| 41 | # quantization |
| 42 | bitsandbytes==0.50.0 |
| 43 | buildkite-test-collector==0.1.9 |
| 44 | tilelang==0.1.10 |
| 45 | |
| 46 | genai_perf>=0.0.8 |
| 47 | tritonclient>=2.51.0 |
| 48 | |
| 49 | # The version of gRPC libraries should be consistent with each other |
| 50 | grpcio==1.78.0 |
| 51 | grpcio-reflection==1.78.0 |
| 52 | |
| 53 | arctic-inference==0.1.1 # Required for suffix decoding test |
| 54 | numba==0.65.0 # Required for N-gram speculative decoding |
| 55 | runai-model-streamer[s3,gcs,azure]==0.15.7 |
| 56 | fastsafetensors>=0.3.3 |
| 57 | instanttensor>=0.1.9 |
| 58 | decord==0.6.0 |
| 59 | |
| 60 | # Prithvi tests |
| 61 | # terratorch is temporarily disabled while PyPI has the `lightning` package |
| 62 | # in `quarantined` status (every published terratorch version transitively |
| 63 | # requires `lightning`, so the resolver fails with "no versions of lightning"). |
| 64 | # Re-enable once PyPI lifts the quarantine. Tracked in #41376. |
| 65 | # terratorch>=1.2.2 |
| 66 | imagehash # Required for Prithvi tests |
| 67 | segmentation-models-pytorch>0.4.0 # Required for Prithvi tests |
| 68 | |
| 69 | gpt-oss[eval]>=0.0.7; python_version > '3.11' |
| 70 | |
| 71 | perceptron # required for isaac test |
| 72 | kaldi-native-fbank>=1.18.7 # required for fireredasr2 test |
| 73 | cohere_melody>=0.11.1 # required for cohere command reasoning/tool parser tests |
| 74 | cohere>=7.0.0 # required for cohere chat v2 api tests (protocol/serving import cohere.types) |
| 75 | |
| 76 | # Newer versions of datasets require torchcoded, that makes the tests fail in CI because of a missing library. |
| 77 | # Older versions are in conflict with terratorch requirements. |
| 78 | datasets>=3.3.0,<=3.6.0 |
| 79 | |
| 80 | openpyxl # required for perf comparison excel report |
| 81 | plotly # required for perf comparison html report |
| 82 | |
| 83 | # Dependabot GHSA floors for transitive test deps (excl. dependabot.yml ignores). |
| 84 | # Skipped (no patched release yet): sqlitedict, accelerate, lmcache, nltk GHSA-8mgp-746c-j5xp. |
| 85 | urllib3>=2.7.0 |
| 86 | cryptography>=50.0.0 |
| 87 | pyjwt>=2.14.0 |
| 88 | pyasn1>=0.6.4 |
| 89 | python-multipart>=0.0.31 |
| 90 | anyio>=4.14.2 |
| 91 | httpx2>=2.12.0 |
| 92 | werkzeug>=3.1.6 |
| 93 | mako>=1.3.12 |
| 94 | black>=26.3.1 |
| 95 | msgpack>=1.2.1 |
| 96 | lxml>=6.1.0 |
| 97 | pyarrow>=23.0.1 |
| 98 | orjson>=3.11.6 |
| 99 | hpack>=4.2.0 |
| 100 | h2>=4.4.1 |
| 101 | pydantic-settings>=2.14.2 |
| 102 | virtualenv>=20.36.1 |
| 103 | fonttools>=4.60.2 |
| 104 | pygments>=2.20.0 |
| 105 | |
| 106 | # ROCm-specific extras (not in CUDA cuda.in) |
| 107 | rapidfuzz |
| 108 | # torchgeo also pulled in `lightning` transitively; disabled for the same |
| 109 | # quarantine reason as terratorch above. Restore once the quarantine clears. |
| 110 | # torchgeo==0.7.0 |
| 111 | multiprocess==0.70.16 |