rocm-test.txt 1.81 KB
Newer Older
1
2
# Common dependencies
-r common.txt
3
4

# Test infrastructure
5
tblib==3.1.0
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
pytest==8.3.5
pytest-asyncio==0.24.0
pytest-timeout==2.3.1
pytest-cov==6.3.0
pytest-forked==1.6.0
pytest-rerunfailures==14.0
pytest-shard==0.1.2

# Async/HTTP dependencies
anyio==4.6.2.post1
    # via httpx, starlette
aiohttp==3.13.0
    # via gpt-oss
httpx==0.27.2
    # HTTP testing
21

22
# Audio processing dependencies
23
audioread==3.0.1
24
    # via librosa
25
cffi==1.17.1
26
    # via soundfile
27
decorator==5.2.1
28
    # via librosa
29
lazy-loader==0.4
30
    # via librosa
31
platformdirs==4.3.6
32
    # via pooch
33
pooch==1.8.2
34
    # via librosa
35
soundfile==0.13.1
36
    # via librosa
37
soxr==0.5.0.post1
38
    # via librosa
39
40
librosa==0.10.2.post1

41
42
43
44
45
# Retrieval and search
bm25s==0.2.13
    # via mteb
pystemmer==3.0.0
    # via mteb
46

47
# Multi-modal processing
48
blobfile==3.0.0
49
50
51
    # Multi-Modal Models Test
decord==0.6.0
    # video processing, required by entrypoints/openai/test_video.py
52
rapidfuzz==3.12.1
53

54
55
56
# OpenAI compatibility and testing
gpt-oss==0.0.8
    # OpenAI compatibility tests
57
schemathesis==3.39.15
58
    # OpenAI schema test
59

60
# Evaluation and benchmarking
61
lm-eval[api]>=0.4.9.2
62
jiwer==4.0.0
63

64
# Required for multiprocessed tests that use spawn method, Datasets and Evaluate Test
65
66
multiprocess==0.70.16

67
68
69
# Required for v1/metrics/test_engine_logger_apis.py
ray[cgraph,default]>=2.48.0

70
71
# Plugins test
terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e
72
torchgeo==0.7.0
73
74
75
76
77
    # via terratorch
# MTEB Benchmark Test
mteb==2.1.2

# Data processing
78
xgrammar @ git+https://github.com/divakar-amd/xgrammar@3272f7c520564858056a60480d5afdf69ae79c84
79
# Test async scheduling
80
81
82
83
84
85

# Utilities
num2words==0.5.14
    # via lm-eval
pqdm==0.2.0
    # via lm-eval
86
87

# Required for suffix decoding test
88
arctic-inference == 0.1.1
89
90
# Required for Nemotron test
open-clip-torch==2.32.0