-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Expand file tree
/
Copy pathrequirements-dev.txt
More file actions
75 lines (75 loc) · 2.27 KB
/
Copy pathrequirements-dev.txt
File metadata and controls
75 lines (75 loc) · 2.27 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
-r requirements.txt
# Pin the exact diffusers version VisualGen LPIPS tests run against; install it
# before pytest collection so already-imported Diffusers modules match the
# package files. 0.39.0 matches the runtime floor in requirements.txt.
diffusers==0.39.0
# get_sysinfo.py's Linux-distribution probe. Must be explicit: it used to arrive
# transitively via openai, and openai 3.3.1 dropped it, which silently emptied
# every sysinfo-rendered CI test list (linux_distribution_name became 'na').
distro==1.9.0
# Imported directly by CI/infra Python but otherwise only present transitively
# (requests via datasets/tiktoken; pyyaml via pydantic-settings[yaml]). Declared
# explicitly so infra never depends on a third-party's transitive dep set --
# the distro failure mode above. Ranges, NOT == (unlike distro, which is pip-
# only): the CI base image ships these as Debian packages (python3-yaml,
# python3-requests) with no pip RECORD, so an exact pin forces an uninstall pip
# cannot perform (uninstall-no-record-file). Floors are the real requirement
# (requests>=2.32.4 also clears CVE-2024-47081); ceilings block unvetted
# releases -- pyyaml<6.0.3 excludes 6.0.3, which fails to import on Python 3.12
# (pyyaml issue #891).
pyyaml>=6.0.1,<6.0.3
requests>=2.32.4,<3
boto3
einops
# Reference implementation for KDA kernel parity tests.
flash-linear-attention==0.5.2
lpips
graphviz
mypy==1.19.1
mako
oyaml
parameterized
pre-commit
pybind11
pybind11-stubgen
# 8.4+ needed for `pytest.RaisesExc` support in `pytest.mark.xfail`.
pytest>=8.4,<9.1
pytest-asyncio
pytest-cov
pytest-csv
pytest-env
pytest-forked
pytest-xdist
pytest-timeout
pytest-split
pytest-mock
pytest-threadleak
pytest-unused-fixtures
rouge_score
cloudpickle
typeguard==4.4.4
typing-extensions==4.15.0
bandit==1.7.7
jsonlines==4.0.0
jieba==0.42.1
rouge==1.0.1
pytest-rerunfailures
ruff==0.9.4
lm_eval[api]==0.4.10
docstring_parser
genai-perf==0.0.13
opentelemetry-sdk>=1.26.0
opentelemetry-api>=1.26.0
opentelemetry-exporter-otlp>=1.26.0
opentelemetry-semantic-conventions-ai>=0.4.1
rapidfuzz==3.14.5
aiperf==0.8.0
nanobind>=2.9.0
nixl-cu13==1.4.0
cupti-python>=13.0,<13.2
nvidia-cuda-cupti>=13.0,<13.2
cxxfilt
hf-transfer==0.1.9
line_profiler
# workaround to prevent picking up 20260602.2.5 which pulls in NumPy 2.5.0
numpy-typing-compat!=20260602.2.5