main
in 83 lines 1.77 KB
Raw
1 -r ../common.txt
2
3 # --- Test Infrastructure ---
4 tblib==3.1.0
5 pytest
6 pytest_asyncio
7 pytest-timeout
8 pytest-cov
9 pytest-forked
10 pytest-rerunfailures
11 pytest-shard
12
13 # --- Core Tools & Bindings ---
14
15 absl-py
16 accelerate
17 arctic-inference
18 lm_eval[api]>=0.4.12
19 nltk>=3.10.3 # pin transitive rouge-score/lm-eval dep for known GHSAs
20 modelscope<1.38
21 runai-model-streamer[s3,gcs,azure]>=0.15.7
22 tensorizer==2.10.1
23 transformers==5.17.0
24
25 # --- Audio Processing ---
26 av
27 librosa
28 audioread
29 soxr
30 pooch
31 soundfile
32
33 # --- Tool Parsing & Evaluation ---
34 blobfile
35 rapidfuzz
36 gpt-oss[eval] >= 0.0.7; python_version > '3.11'
37 schemathesis>=4.0.0 # Required for openai schema test.
38 jiwer
39 bm25s
40 pystemmer
41 mteb[bm25s]>=2, <3 # required for mteb test
42 sentence-transformers==5.2.0 # required for embedding tests
43 num2words
44 pqdm
45
46 # --- Vision & Multimodal ---
47 timm>=1.0.17 # required for internvl and gemma3n-mm test
48 albumentations
49 ftfy
50 matplotlib
51 decord==0.6.0; platform_machine == "x86_64"
52 mistral-common[image,audio]>=1.11.6 # required for voxtral test
53
54 open_clip_torch==2.32.0 # Required for nemotron_vl test, Nemotron Parse in test_common.py
55
56 # --- Observability ---
57 opentelemetry-api>=1.26.0
58 opentelemetry-exporter-otlp>=1.26.0
59 opentelemetry-sdk>=1.26.0
60 opentelemetry-semantic-conventions-ai>=0.4.1
61
62 cohere_melody>=0.11.1 # required for cohere command reasoning/tool parser tests
63 cohere>=7.0.0 # required for cohere chat v2 api tests (protocol/serving import cohere.types)
64
65 # Dependabot GHSA floors for transitive test deps (excl. dependabot.yml ignores).
66 urllib3>=2.7.0
67 cryptography>=50.0.0
68 pyjwt>=2.14.0
69 pyasn1>=0.6.4
70 python-multipart>=0.0.31
71 anyio>=4.14.2
72 httpx2>=2.12.0
73 werkzeug>=3.1.6
74 mako>=1.3.12
75 msgpack>=1.2.1
76 lxml>=6.1.0
77 pyarrow>=23.0.1
78 orjson>=3.11.6
79 hpack>=4.2.0
80 h2>=4.4.1
81 pydantic-settings>=2.14.2
82 fonttools>=4.60.2
83 pygments>=2.20.0