Skip to content

GitLab

"docs/pages/components/vscode:/vscode.git/clone" did not exist on "233a1e9ad0bf012cc6e16d557c8d2bf7ed41ae91"
Unverified Commit fc645f1a authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Add structure to `requirements/` directory (#39024) 


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: default avatarRobert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
parent 2d80cf9d
Hide whitespace changes
Inline Side-by-side
Showing with 273 additions and 267 deletions
docs/getting_started/installation/cpu.arm.inc.md
View file @ fc645f1a
...@@ -96,14 +96,14 @@ cd vllm_source ...@@ -96,14 +96,14 @@ cd vllm_source
Third, install required dependencies: Third, install required dependencies:
```bash ```bash
uv pip install -r requirements/cpu-build.txt --torch-backend cpu uv pip install -r requirements/build/cpu.txt --torch-backend cpu
uv pip install -r requirements/cpu.txt --torch-backend cpu uv pip install -r requirements/cpu.txt --torch-backend cpu
``` ```
??? console "pip" ??? console "pip"
```bash ```bash
pip install --upgrade pip pip install --upgrade pip
pip install -v -r requirements/cpu-build.txt --extra-index-url https://download.pytorch.org/whl/cpu pip install -v -r requirements/build/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
``` ```
......
docs/getting_started/installation/cpu.s390x.inc.md
View file @ fc645f1a
...@@ -46,10 +46,10 @@ Execute the following commands to build and install vLLM from source. ...@@ -46,10 +46,10 @@ Execute the following commands to build and install vLLM from source.
Please build the following dependencies, `torchvision`, `pyarrow` from source before building vLLM. Please build the following dependencies, `torchvision`, `pyarrow` from source before building vLLM.
```bash ```bash
sed -i '/^torch/d' requirements/build.txt # remove torch from requirements/build.txt since we use nightly builds sed -i '/^torch/d' requirements/build/cuda.txt # remove torch from requirements/build/cuda.txt since we use nightly builds
uv pip install -v \ uv pip install -v \
--torch-backend auto \ --torch-backend auto \
-r requirements/build.txt \ -r requirements/build/cuda.txt \
-r requirements/cpu.txt \ -r requirements/cpu.txt \
VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \ VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \
uv pip install dist/*.whl uv pip install dist/*.whl
...@@ -57,10 +57,10 @@ Execute the following commands to build and install vLLM from source. ...@@ -57,10 +57,10 @@ Execute the following commands to build and install vLLM from source.
??? console "pip" ??? console "pip"
```bash ```bash
sed -i '/^torch/d' requirements/build.txt # remove torch from requirements/build.txt since we use nightly builds sed -i '/^torch/d' requirements/build/cuda.txt # remove torch from requirements/build/cuda.txt since we use nightly builds
pip install -v \ pip install -v \
--extra-index-url https://download.pytorch.org/whl/nightly/cpu \ --extra-index-url https://download.pytorch.org/whl/nightly/cpu \
-r requirements/build.txt \ -r requirements/build/cuda.txt \
-r requirements/cpu.txt \ -r requirements/cpu.txt \
VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \ VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \
pip install dist/*.whl pip install dist/*.whl
......
docs/getting_started/installation/cpu.x86.inc.md
View file @ fc645f1a
...@@ -88,14 +88,14 @@ cd vllm_source ...@@ -88,14 +88,14 @@ cd vllm_source
Install the required dependencies: Install the required dependencies:
```bash ```bash
uv pip install -r requirements/cpu-build.txt --torch-backend cpu uv pip install -r requirements/build/cpu.txt --torch-backend cpu
uv pip install -r requirements/cpu.txt --torch-backend cpu uv pip install -r requirements/cpu.txt --torch-backend cpu
``` ```
??? console "pip" ??? console "pip"
```bash ```bash
pip install --upgrade pip pip install --upgrade pip
pip install -v -r requirements/cpu-build.txt --extra-index-url https://download.pytorch.org/whl/cpu pip install -v -r requirements/build/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
``` ```
......
docs/getting_started/installation/gpu.cuda.inc.md
View file @ fc645f1a
...@@ -162,7 +162,7 @@ To build vLLM using an existing PyTorch installation: ...@@ -162,7 +162,7 @@ To build vLLM using an existing PyTorch installation:
git clone https://github.com/vllm-project/vllm.git git clone https://github.com/vllm-project/vllm.git
cd vllm cd vllm
python use_existing_torch.py python use_existing_torch.py
uv pip install -r requirements/build.txt uv pip install -r requirements/build/cuda.txt
uv pip install --no-build-isolation -e . uv pip install --no-build-isolation -e .
``` ```
......
docs/mkdocs/hooks/generate_argparse.py
View file @ fc645f1a
...@@ -46,7 +46,7 @@ mock_if_no_torch( ...@@ -46,7 +46,7 @@ mock_if_no_torch(
# Mock any version checks by reading from compiled CI requirements # Mock any version checks by reading from compiled CI requirements
with open(ROOT_DIR / "requirements/test.txt") as f: with open(ROOT_DIR / "requirements/test/cuda.txt") as f:
VERSIONS = dict(line.strip().split("==") for line in f if "==" in line) VERSIONS = dict(line.strip().split("==") for line in f if "==" in line)
importlib.metadata.version = lambda name: VERSIONS.get(name) or "0.0.0" importlib.metadata.version = lambda name: VERSIONS.get(name) or "0.0.0"
......
pyproject.toml
View file @ fc645f1a
[build-system] [build-system]
# Should be mirrored in requirements/build.txt # Should be mirrored in requirements/build/cuda.txt
requires = [ requires = [
"cmake>=3.26.1", "cmake>=3.26.1",
"ninja", "ninja",
......
requirements/rocm-build.txt → requirements/build/rocm.txt
View file @ fc645f1a
# Common dependencies # Common dependencies
-r common.txt -r ../common.txt
--extra-index-url https://download.pytorch.org/whl/rocm7.1 --extra-index-url https://download.pytorch.org/whl/rocm7.1
......
requirements/dev.txt
View file @ fc645f1a
-r lint.txt -r lint.txt
-r test.txt -r test/cuda.txt
# Avoid adding requirements directly to this file. # Avoid adding requirements directly to this file.
# Instead, modify the two files referenced above. # Instead, modify the two files referenced above.
requirements/test.txt → requirements/test/cuda.txt
View file @ fc645f1a
# This file was autogenerated by uv via the following command: # This file was autogenerated by uv via the following command:
# uv pip compile requirements/test.in -c requirements/common.txt -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu130 --python-platform x86_64-manylinux_2_28 --python-version 3.12 # uv pip compile requirements/test/cuda.in -c requirements/cuda.txt -o requirements/test/cuda.txt --index-strategy unsafe-best-match --torch-backend cu130 --python-platform x86_64-manylinux_2_28 --python-version 3.12
absl-py==2.1.0 absl-py==2.1.0
# via # via
# rouge-score # rouge-score
...@@ -29,7 +29,7 @@ albucore==0.0.16 ...@@ -29,7 +29,7 @@ albucore==0.0.16
# via terratorch # via terratorch
albumentations==1.4.6 albumentations==1.4.6
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# terratorch # terratorch
alembic==1.16.4 alembic==1.16.4
# via optuna # via optuna
...@@ -46,7 +46,7 @@ anyio==4.6.2.post1 ...@@ -46,7 +46,7 @@ anyio==4.6.2.post1
# httpx # httpx
# starlette # starlette
arctic-inference==0.1.1 arctic-inference==0.1.1
# via -r requirements/test.in # via -r requirements/test/cuda.in
argcomplete==3.5.1 argcomplete==3.5.1
# via datamodel-code-generator # via datamodel-code-generator
arrow==1.3.0 arrow==1.3.0
...@@ -64,7 +64,7 @@ attrs==24.2.0 ...@@ -64,7 +64,7 @@ attrs==24.2.0
audioread==3.0.1 audioread==3.0.1
# via librosa # via librosa
av==16.1.0 av==16.1.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
azure-core==1.38.2 azure-core==1.38.2
# via # via
# azure-identity # azure-identity
...@@ -75,16 +75,16 @@ azure-storage-blob==12.28.0 ...@@ -75,16 +75,16 @@ azure-storage-blob==12.28.0
# via runai-model-streamer-azure # via runai-model-streamer-azure
backoff==2.2.1 backoff==2.2.1
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# schemathesis # schemathesis
bitsandbytes==0.49.2 bitsandbytes==0.49.2
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# lightning # lightning
black==24.10.0 black==24.10.0
# via datamodel-code-generator # via datamodel-code-generator
blobfile==3.0.0 blobfile==3.0.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
bm25s==0.2.13 bm25s==0.2.13
# via mteb # via mteb
boto3==1.35.57 boto3==1.35.57
...@@ -98,7 +98,7 @@ botocore==1.35.57 ...@@ -98,7 +98,7 @@ botocore==1.35.57
bounded-pool-executor==0.0.3 bounded-pool-executor==0.0.3
# via pqdm # via pqdm
buildkite-test-collector==0.1.9 buildkite-test-collector==0.1.9
# via -r requirements/test.in # via -r requirements/test/cuda.in
cachetools==5.5.2 cachetools==5.5.2
# via google-auth # via google-auth
certifi==2024.8.30 certifi==2024.8.30
...@@ -176,21 +176,21 @@ cupy-cuda12x==13.6.0 ...@@ -176,21 +176,21 @@ cupy-cuda12x==13.6.0
cycler==0.12.1 cycler==0.12.1
# via matplotlib # via matplotlib
datamodel-code-generator==0.26.3 datamodel-code-generator==0.26.3
# via -r requirements/test.in # via -r requirements/test/cuda.in
dataproperty==1.0.1 dataproperty==1.0.1
# via # via
# pytablewriter # pytablewriter
# tabledata # tabledata
datasets==3.3.0 datasets==3.3.0
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# evaluate # evaluate
# lm-eval # lm-eval
# mteb # mteb
decorator==5.1.1 decorator==5.1.1
# via librosa # via librosa
decord==0.6.0 decord==0.6.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
diffusers==0.36.0 diffusers==0.36.0
# via terratorch # via terratorch
dill==0.3.8 dill==0.3.8
...@@ -211,7 +211,7 @@ docstring-parser==0.17.0 ...@@ -211,7 +211,7 @@ docstring-parser==0.17.0
# via jsonargparse # via jsonargparse
einops==0.8.1 einops==0.8.1
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# encodec # encodec
# terratorch # terratorch
# torchgeo # torchgeo
...@@ -236,7 +236,9 @@ fastparquet==2024.11.0 ...@@ -236,7 +236,9 @@ fastparquet==2024.11.0
fastrlock==0.8.2 fastrlock==0.8.2
# via cupy-cuda12x # via cupy-cuda12x
fastsafetensors==0.2.2 fastsafetensors==0.2.2
# via -r requirements/test.in # via
# -c requirements/cuda.txt
# -r requirements/test/cuda.in
filelock==3.16.1 filelock==3.16.1
# via # via
# -c requirements/common.txt # -c requirements/common.txt
...@@ -273,7 +275,7 @@ fsspec==2024.12.0 ...@@ -273,7 +275,7 @@ fsspec==2024.12.0
ftfy==6.3.1 ftfy==6.3.1
# via open-clip-torch # via open-clip-torch
genai-perf==0.0.16 genai-perf==0.0.16
# via -r requirements/test.in # via -r requirements/test/cuda.in
genson==1.3.0 genson==1.3.0
# via datamodel-code-generator # via datamodel-code-generator
geopandas==1.0.1 geopandas==1.0.1
...@@ -306,19 +308,19 @@ google-resumable-media==2.7.2 ...@@ -306,19 +308,19 @@ google-resumable-media==2.7.2
googleapis-common-protos==1.70.0 googleapis-common-protos==1.70.0
# via google-api-core # via google-api-core
gpt-oss==0.0.8 gpt-oss==0.0.8
# via -r requirements/test.in # via -r requirements/test/cuda.in
graphql-core==3.2.6 graphql-core==3.2.6
# via hypothesis-graphql # via hypothesis-graphql
greenlet==3.2.3 greenlet==3.2.3
# via sqlalchemy # via sqlalchemy
grpcio==1.78.0 grpcio==1.78.0
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# grpcio-reflection # grpcio-reflection
# ray # ray
# tensorboard # tensorboard
grpcio-reflection==1.78.0 grpcio-reflection==1.78.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
h11==0.14.0 h11==0.14.0
# via # via
# httpcore # httpcore
...@@ -341,7 +343,7 @@ httpcore==1.0.6 ...@@ -341,7 +343,7 @@ httpcore==1.0.6
# via httpx # via httpx
httpx==0.27.2 httpx==0.27.2
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# diffusers # diffusers
# perceptron # perceptron
# schemathesis # schemathesis
...@@ -386,7 +388,7 @@ idna==3.10 ...@@ -386,7 +388,7 @@ idna==3.10
# requests # requests
# yarl # yarl
imagehash==4.3.2 imagehash==4.3.2
# via -r requirements/test.in # via -r requirements/test/cuda.in
imageio==2.37.0 imageio==2.37.0
# via scikit-image # via scikit-image
importlib-metadata==8.7.0 importlib-metadata==8.7.0
...@@ -400,7 +402,7 @@ inflect==5.6.2 ...@@ -400,7 +402,7 @@ inflect==5.6.2
iniconfig==2.0.0 iniconfig==2.0.0
# via pytest # via pytest
instanttensor==0.1.5 instanttensor==0.1.5
# via -r requirements/test.in # via -r requirements/test/cuda.in
isodate==0.7.2 isodate==0.7.2
# via azure-storage-blob # via azure-storage-blob
isoduration==20.11.0 isoduration==20.11.0
...@@ -414,7 +416,7 @@ jinja2==3.1.6 ...@@ -414,7 +416,7 @@ jinja2==3.1.6
# lm-eval # lm-eval
# torch # torch
jiwer==3.0.5 jiwer==3.0.5
# via -r requirements/test.in # via -r requirements/test/cuda.in
jmespath==1.0.1 jmespath==1.0.1
# via # via
# boto3 # boto3
...@@ -445,7 +447,7 @@ jsonschema-specifications==2024.10.1 ...@@ -445,7 +447,7 @@ jsonschema-specifications==2024.10.1
junit-xml==1.9 junit-xml==1.9
# via schemathesis # via schemathesis
kaldi-native-fbank==1.22.3 kaldi-native-fbank==1.22.3
# via -r requirements/test.in # via -r requirements/test/cuda.in
kaleido==0.2.1 kaleido==0.2.1
# via genai-perf # via genai-perf
kiwisolver==1.4.7 kiwisolver==1.4.7
...@@ -461,7 +463,7 @@ lazy-loader==0.4 ...@@ -461,7 +463,7 @@ lazy-loader==0.4
libnacl==2.1.0 libnacl==2.1.0
# via tensorizer # via tensorizer
librosa==0.10.2.post1 librosa==0.10.2.post1
# via -r requirements/test.in # via -r requirements/test/cuda.in
lightly==1.5.22 lightly==1.5.22
# via # via
# terratorch # terratorch
...@@ -480,7 +482,7 @@ lightning-utilities==0.14.3 ...@@ -480,7 +482,7 @@ lightning-utilities==0.14.3
llvmlite==0.44.0 llvmlite==0.44.0
# via numba # via numba
lm-eval==0.4.11 lm-eval==0.4.11
# via -r requirements/test.in # via -r requirements/test/cuda.in
lxml==5.3.0 lxml==5.3.0
# via # via
# blobfile # blobfile
...@@ -499,7 +501,7 @@ markupsafe==3.0.1 ...@@ -499,7 +501,7 @@ markupsafe==3.0.1
# werkzeug # werkzeug
matplotlib==3.9.2 matplotlib==3.9.2
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# lightning # lightning
# pycocotools # pycocotools
# torchgeo # torchgeo
...@@ -513,7 +515,7 @@ mdurl==0.1.2 ...@@ -513,7 +515,7 @@ mdurl==0.1.2
mistral-common==1.11.0 mistral-common==1.11.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/test.in # -r requirements/test/cuda.in
more-itertools==10.5.0 more-itertools==10.5.0
# via lm-eval # via lm-eval
mpmath==1.3.0 mpmath==1.3.0
...@@ -529,7 +531,7 @@ msgpack==1.1.0 ...@@ -529,7 +531,7 @@ msgpack==1.1.0
# librosa # librosa
# ray # ray
mteb==2.8.3 mteb==2.8.3
# via -r requirements/test.in # via -r requirements/test/cuda.in
multidict==6.1.0 multidict==6.1.0
# via # via
# aiohttp # aiohttp
...@@ -547,15 +549,16 @@ networkx==3.2.1 ...@@ -547,15 +549,16 @@ networkx==3.2.1
nltk==3.9.1 nltk==3.9.1
# via rouge-score # via rouge-score
num2words==0.5.14 num2words==0.5.14
# via -r requirements/test.in # via -r requirements/test/cuda.in
numba==0.61.2 numba==0.61.2
# via # via
# -r requirements/test.in # -c requirements/cuda.txt
# -r requirements/test/cuda.in
# librosa # librosa
# resampy # resampy
numpy==2.2.6 numpy==2.2.6
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# accelerate # accelerate
# albucore # albucore
# albumentations # albumentations
...@@ -661,7 +664,7 @@ omegaconf==2.3.0 ...@@ -661,7 +664,7 @@ omegaconf==2.3.0
# hydra-core # hydra-core
# lightning # lightning
open-clip-torch==2.32.0 open-clip-torch==2.32.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
openai-harmony==0.0.4 openai-harmony==0.0.4
# via # via
# -c requirements/common.txt # -c requirements/common.txt
...@@ -673,12 +676,12 @@ opencensus-context==0.1.3 ...@@ -673,12 +676,12 @@ opencensus-context==0.1.3
opencv-python-headless==4.13.0.90 opencv-python-headless==4.13.0.90
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/test.in # -r requirements/test/cuda.in
# albucore # albucore
# albumentations # albumentations
# mistral-common # mistral-common
openpyxl==3.1.5 openpyxl==3.1.5
# via -r requirements/test.in # via -r requirements/test/cuda.in
opentelemetry-api==1.35.0 opentelemetry-api==1.35.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
...@@ -754,9 +757,9 @@ pathvalidate==3.2.1 ...@@ -754,9 +757,9 @@ pathvalidate==3.2.1
patsy==1.0.1 patsy==1.0.1
# via statsmodels # via statsmodels
peft==0.16.0 peft==0.16.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
perceptron==0.1.4 perceptron==0.1.4
# via -r requirements/test.in # via -r requirements/test/cuda.in
perf-analyzer==0.1.0 perf-analyzer==0.1.0
# via genai-perf # via genai-perf
pillow==10.4.0 pillow==10.4.0
...@@ -782,7 +785,7 @@ platformdirs==4.3.6 ...@@ -782,7 +785,7 @@ platformdirs==4.3.6
# wandb # wandb
plotly==5.24.1 plotly==5.24.1
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# genai-perf # genai-perf
pluggy==1.5.0 pluggy==1.5.0
# via # via
...@@ -795,7 +798,7 @@ pooch==1.8.2 ...@@ -795,7 +798,7 @@ pooch==1.8.2
portalocker==2.10.1 portalocker==2.10.1
# via sacrebleu # via sacrebleu
pqdm==0.2.0 pqdm==0.2.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
prometheus-client==0.22.0 prometheus-client==0.22.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
...@@ -852,7 +855,7 @@ pycryptodomex==3.22.0 ...@@ -852,7 +855,7 @@ pycryptodomex==3.22.0
pydantic==2.12.0 pydantic==2.12.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/test.in # -r requirements/test/cuda.in
# albumentations # albumentations
# datamodel-code-generator # datamodel-code-generator
# fastapi # fastapi
...@@ -891,7 +894,7 @@ pytablewriter==1.2.0 ...@@ -891,7 +894,7 @@ pytablewriter==1.2.0
# via lm-eval # via lm-eval
pytest==8.3.5 pytest==8.3.5
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# buildkite-test-collector # buildkite-test-collector
# genai-perf # genai-perf
# pytest-asyncio # pytest-asyncio
...@@ -904,21 +907,21 @@ pytest==8.3.5 ...@@ -904,21 +907,21 @@ pytest==8.3.5
# pytest-timeout # pytest-timeout
# schemathesis # schemathesis
pytest-asyncio==0.24.0 pytest-asyncio==0.24.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
pytest-cov==6.3.0 pytest-cov==6.3.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
pytest-forked==1.6.0 pytest-forked==1.6.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
pytest-mock==3.14.0 pytest-mock==3.14.0
# via genai-perf # via genai-perf
pytest-rerunfailures==14.0 pytest-rerunfailures==14.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
pytest-shard==0.1.2 pytest-shard==0.1.2
# via -r requirements/test.in # via -r requirements/test/cuda.in
pytest-subtests==0.14.1 pytest-subtests==0.14.1
# via schemathesis # via schemathesis
pytest-timeout==2.3.1 pytest-timeout==2.3.1
# via -r requirements/test.in # via -r requirements/test/cuda.in
python-box==7.3.2 python-box==7.3.2
# via terratorch # via terratorch
python-dateutil==2.9.0.post0 python-dateutil==2.9.0.post0
...@@ -972,7 +975,7 @@ rasterio==1.4.3 ...@@ -972,7 +975,7 @@ rasterio==1.4.3
# terratorch # terratorch
# torchgeo # torchgeo
ray==2.48.0 ray==2.48.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
redis==5.2.0 redis==5.2.0
# via tensorizer # via tensorizer
referencing==0.35.1 referencing==0.35.1
...@@ -1015,7 +1018,7 @@ requests==2.32.3 ...@@ -1015,7 +1018,7 @@ requests==2.32.3
# transformers # transformers
# wandb # wandb
resampy==0.4.3 resampy==0.4.3
# via -r requirements/test.in # via -r requirements/test/cuda.in
responses==0.25.3 responses==0.25.3
# via genai-perf # via genai-perf
rfc3339-validator==0.1.4 rfc3339-validator==0.1.4
...@@ -1043,7 +1046,7 @@ rsa==4.9.1 ...@@ -1043,7 +1046,7 @@ rsa==4.9.1
rtree==1.4.0 rtree==1.4.0
# via torchgeo # via torchgeo
runai-model-streamer==0.15.7 runai-model-streamer==0.15.7
# via -r requirements/test.in # via -r requirements/test/cuda.in
runai-model-streamer-azure==0.15.7 runai-model-streamer-azure==0.15.7
# via runai-model-streamer # via runai-model-streamer
runai-model-streamer-gcs==0.15.7 runai-model-streamer-gcs==0.15.7
...@@ -1064,7 +1067,7 @@ safetensors==0.4.5 ...@@ -1064,7 +1067,7 @@ safetensors==0.4.5
# timm # timm
# transformers # transformers
schemathesis==3.39.15 schemathesis==3.39.15
# via -r requirements/test.in # via -r requirements/test/cuda.in
scikit-image==0.25.2 scikit-image==0.25.2
# via # via
# albumentations # albumentations
...@@ -1091,12 +1094,12 @@ scipy==1.13.1 ...@@ -1091,12 +1094,12 @@ scipy==1.13.1
# vocos # vocos
segmentation-models-pytorch==0.5.0 segmentation-models-pytorch==0.5.0
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# terratorch # terratorch
# torchgeo # torchgeo
sentence-transformers==5.2.0 sentence-transformers==5.2.0
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# mteb # mteb
sentry-sdk==2.52.0 sentry-sdk==2.52.0
# via wandb # via wandb
...@@ -1136,7 +1139,7 @@ sortedcontainers==2.4.0 ...@@ -1136,7 +1139,7 @@ sortedcontainers==2.4.0
# via hypothesis # via hypothesis
soundfile==0.12.1 soundfile==0.12.1
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# genai-perf # genai-perf
# librosa # librosa
# mistral-common # mistral-common
...@@ -1172,7 +1175,7 @@ tabulate==0.9.0 ...@@ -1172,7 +1175,7 @@ tabulate==0.9.0
tacoreader==0.5.6 tacoreader==0.5.6
# via terratorch # via terratorch
tblib==3.1.0 tblib==3.1.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
tcolorpy==0.1.6 tcolorpy==0.1.6
# via pytablewriter # via pytablewriter
tenacity==9.1.2 tenacity==9.1.2
...@@ -1187,13 +1190,13 @@ tensorboard-data-server==0.7.2 ...@@ -1187,13 +1190,13 @@ tensorboard-data-server==0.7.2
tensorboardx==2.6.4 tensorboardx==2.6.4
# via lightning # via lightning
tensorizer==2.10.1 tensorizer==2.10.1
# via -r requirements/test.in # via -r requirements/test/cuda.in
termcolor==3.1.0 termcolor==3.1.0
# via # via
# gpt-oss # gpt-oss
# terratorch # terratorch
terratorch==1.2.2 terratorch==1.2.2
# via -r requirements/test.in # via -r requirements/test/cuda.in
threadpoolctl==3.5.0 threadpoolctl==3.5.0
# via scikit-learn # via scikit-learn
tifffile==2025.3.30 tifffile==2025.3.30
...@@ -1208,7 +1211,7 @@ tiktoken==0.12.0 ...@@ -1208,7 +1211,7 @@ tiktoken==0.12.0
# mistral-common # mistral-common
timm==1.0.17 timm==1.0.17
# via # via
# -r requirements/test.in # -r requirements/test/cuda.in
# open-clip-torch # open-clip-torch
# segmentation-models-pytorch # segmentation-models-pytorch
# terratorch # terratorch
...@@ -1216,7 +1219,7 @@ timm==1.0.17 ...@@ -1216,7 +1219,7 @@ timm==1.0.17
tokenizers==0.22.0 tokenizers==0.22.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/test.in # -r requirements/test/cuda.in
# transformers # transformers
tomli==2.2.1 tomli==2.2.1
# via schemathesis # via schemathesis
...@@ -1224,7 +1227,8 @@ tomli-w==1.2.0 ...@@ -1224,7 +1227,8 @@ tomli-w==1.2.0
# via schemathesis # via schemathesis
torch==2.11.0+cu130 torch==2.11.0+cu130
# via # via
# -r requirements/test.in # -c requirements/cuda.txt
# -r requirements/test/cuda.in
# accelerate # accelerate
# bitsandbytes # bitsandbytes
# encodec # encodec
...@@ -1249,7 +1253,8 @@ torch==2.11.0+cu130 ...@@ -1249,7 +1253,8 @@ torch==2.11.0+cu130
# vocos # vocos
torchaudio==2.11.0+cu130 torchaudio==2.11.0+cu130
# via # via
# -r requirements/test.in # -c requirements/cuda.txt
# -r requirements/test/cuda.in
# encodec # encodec
# vocos # vocos
torchgeo==0.7.0 torchgeo==0.7.0
...@@ -1262,7 +1267,8 @@ torchmetrics==1.7.4 ...@@ -1262,7 +1267,8 @@ torchmetrics==1.7.4
# torchgeo # torchgeo
torchvision==0.26.0+cu130 torchvision==0.26.0+cu130
# via # via
# -r requirements/test.in # -c requirements/cuda.txt
# -r requirements/test/cuda.in
# lightly # lightly
# open-clip-torch # open-clip-torch
# segmentation-models-pytorch # segmentation-models-pytorch
...@@ -1292,17 +1298,17 @@ tqdm==4.67.3 ...@@ -1292,17 +1298,17 @@ tqdm==4.67.3
transformers==4.57.5 transformers==4.57.5
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/test.in # -r requirements/test/cuda.in
# genai-perf # genai-perf
# peft # peft
# sentence-transformers # sentence-transformers
# transformers-stream-generator # transformers-stream-generator
transformers-stream-generator==0.0.5 transformers-stream-generator==0.0.5
# via -r requirements/test.in # via -r requirements/test/cuda.in
triton==3.6.0 triton==3.6.0
# via torch # via torch
tritonclient==2.64.0 tritonclient==2.64.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
typepy==1.3.2 typepy==1.3.2
# via # via
# dataproperty # dataproperty
...@@ -1371,11 +1377,11 @@ urllib3==2.2.3 ...@@ -1371,11 +1377,11 @@ urllib3==2.2.3
uvicorn==0.35.0 uvicorn==0.35.0
# via gpt-oss # via gpt-oss
vector-quantize-pytorch==1.21.2 vector-quantize-pytorch==1.21.2
# via -r requirements/test.in # via -r requirements/test/cuda.in
virtualenv==20.31.2 virtualenv==20.31.2
# via ray # via ray
vocos==0.1.0 vocos==0.1.0
# via -r requirements/test.in # via -r requirements/test/cuda.in
wandb==0.24.2 wandb==0.24.2
# via terratorch # via terratorch
wcwidth==0.2.13 wcwidth==0.2.13
......
requirements/rocm-test.in → requirements/test/rocm.in
View file @ fc645f1a
-r common.txt -r ../common.txt
# testing # testing
pytest pytest
...@@ -78,7 +78,7 @@ datasets>=3.3.0,<=3.6.0 ...@@ -78,7 +78,7 @@ datasets>=3.3.0,<=3.6.0
openpyxl # required for perf comparison excel report openpyxl # required for perf comparison excel report
plotly # required for perf comparison html report plotly # required for perf comparison html report
# ROCm-specific extras (not in CUDA test.in) # ROCm-specific extras (not in CUDA cuda.in)
rapidfuzz rapidfuzz
torchgeo==0.7.0 torchgeo==0.7.0
multiprocess==0.70.16 multiprocess==0.70.16
......
requirements/rocm-test.txt → requirements/test/rocm.txt
View file @ fc645f1a
# This file was autogenerated by uv via the following command: # This file was autogenerated by uv via the following command:
# uv pip compile requirements/rocm-test.in -o requirements/rocm-test.txt --index-strategy unsafe-best-match -c requirements/rocm.txt --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13 # uv pip compile requirements/test/rocm.in -c requirements/rocm.txt -o requirements/test/rocm.txt --index-strategy unsafe-best-match --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13
absl-py==2.4.0 absl-py==2.4.0
# via # via
# rouge-score # rouge-score
...@@ -15,7 +15,7 @@ aiohappyeyeballs==2.6.1 ...@@ -15,7 +15,7 @@ aiohappyeyeballs==2.6.1
aiohttp==3.13.3 aiohttp==3.13.3
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# aiohttp-cors # aiohttp-cors
# fsspec # fsspec
# gpt-oss # gpt-oss
...@@ -29,7 +29,7 @@ albucore==0.1.2 ...@@ -29,7 +29,7 @@ albucore==0.1.2
# via terratorch # via terratorch
albumentations==1.4.6 albumentations==1.4.6
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# terratorch # terratorch
alembic==1.18.4 alembic==1.18.4
# via optuna # via optuna
...@@ -42,7 +42,7 @@ annotated-types==0.7.0 ...@@ -42,7 +42,7 @@ annotated-types==0.7.0
anthropic==0.89.0 anthropic==0.89.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
antlr4-python3-runtime==4.9.3 antlr4-python3-runtime==4.9.3
# via # via
# hydra-core # hydra-core
...@@ -57,7 +57,7 @@ anyio==4.13.0 ...@@ -57,7 +57,7 @@ anyio==4.13.0
# starlette # starlette
# watchfiles # watchfiles
arctic-inference==0.1.1 arctic-inference==0.1.1
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
argcomplete==3.6.3 argcomplete==3.6.3
# via datamodel-code-generator # via datamodel-code-generator
arrow==1.4.0 arrow==1.4.0
...@@ -76,7 +76,7 @@ attrs==26.1.0 ...@@ -76,7 +76,7 @@ attrs==26.1.0
audioread==3.0.1 audioread==3.0.1
# via librosa # via librosa
av==16.1.0 av==16.1.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
azure-core==1.39.0 azure-core==1.39.0
# via # via
# azure-identity # azure-identity
...@@ -87,18 +87,18 @@ azure-storage-blob==12.28.0 ...@@ -87,18 +87,18 @@ azure-storage-blob==12.28.0
# via runai-model-streamer-azure # via runai-model-streamer-azure
backoff==2.2.1 backoff==2.2.1
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# schemathesis # schemathesis
bitsandbytes==0.49.2 bitsandbytes==0.49.2
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# lightning # lightning
black==26.3.1 black==26.3.1
# via datamodel-code-generator # via datamodel-code-generator
blake3==1.0.8 blake3==1.0.8
# via -r requirements/common.txt # via -r requirements/test/../common.txt
blobfile==3.0.0 blobfile==3.0.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
bm25s==0.2.13 bm25s==0.2.13
# via mteb # via mteb
boto3==1.42.74 boto3==1.42.74
...@@ -112,11 +112,11 @@ botocore==1.42.74 ...@@ -112,11 +112,11 @@ botocore==1.42.74
bounded-pool-executor==0.0.3 bounded-pool-executor==0.0.3
# via pqdm # via pqdm
buildkite-test-collector==0.1.9 buildkite-test-collector==0.1.9
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
cachetools==7.0.5 cachetools==7.0.5
# via -r requirements/common.txt # via -r requirements/test/../common.txt
cbor2==5.9.0 cbor2==5.9.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
certifi==2026.2.25 certifi==2026.2.25
# via # via
# fiona # fiona
...@@ -162,7 +162,7 @@ cligj==0.7.2 ...@@ -162,7 +162,7 @@ cligj==0.7.2
# fiona # fiona
# rasterio # rasterio
cloudpickle==3.1.2 cloudpickle==3.1.2
# via -r requirements/common.txt # via -r requirements/test/../common.txt
colorama==0.4.6 colorama==0.4.6
# via # via
# perceptron # perceptron
...@@ -175,7 +175,7 @@ colorlog==6.10.1 ...@@ -175,7 +175,7 @@ colorlog==6.10.1
compressed-tensors==0.14.0.1 compressed-tensors==0.14.0.1
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
contourpy==1.3.3 contourpy==1.3.3
# via matplotlib # via matplotlib
coverage==7.13.5 coverage==7.13.5
...@@ -192,25 +192,25 @@ cryptography==46.0.0 ...@@ -192,25 +192,25 @@ cryptography==46.0.0
cycler==0.12.1 cycler==0.12.1
# via matplotlib # via matplotlib
datamodel-code-generator==0.55.0 datamodel-code-generator==0.55.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
dataproperty==1.1.0 dataproperty==1.1.0
# via # via
# pytablewriter # pytablewriter
# tabledata # tabledata
datasets==3.6.0 datasets==3.6.0
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# evaluate # evaluate
# lm-eval # lm-eval
# mteb # mteb
decorator==5.2.1 decorator==5.2.1
# via librosa # via librosa
decord==0.6.0 decord==0.6.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
depyf==0.20.0 depyf==0.20.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
diffusers==0.37.0 diffusers==0.37.0
# via terratorch # via terratorch
dill==0.3.8 dill==0.3.8
...@@ -223,7 +223,7 @@ dill==0.3.8 ...@@ -223,7 +223,7 @@ dill==0.3.8
diskcache==5.6.3 diskcache==5.6.3
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
distlib==0.4.0 distlib==0.4.0
# via virtualenv # via virtualenv
distro==1.9.0 distro==1.9.0
...@@ -242,8 +242,8 @@ docstring-parser==0.17.0 ...@@ -242,8 +242,8 @@ docstring-parser==0.17.0
# jsonargparse # jsonargparse
einops==0.8.2 einops==0.8.2
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# encodec # encodec
# terratorch # terratorch
# torchgeo # torchgeo
...@@ -264,7 +264,7 @@ evaluate==0.4.6 ...@@ -264,7 +264,7 @@ evaluate==0.4.6
fastapi==0.135.2 fastapi==0.135.2
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# gpt-oss # gpt-oss
# model-hosting-container-standards # model-hosting-container-standards
fastapi-cli==0.0.24 fastapi-cli==0.0.24
...@@ -278,11 +278,11 @@ fastparquet==2026.3.0 ...@@ -278,11 +278,11 @@ fastparquet==2026.3.0
fastsafetensors==0.2.2 fastsafetensors==0.2.2
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
filelock==3.25.2 filelock==3.25.2
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# blobfile # blobfile
# datasets # datasets
# diffusers # diffusers
...@@ -317,7 +317,7 @@ fsspec==2025.3.0 ...@@ -317,7 +317,7 @@ fsspec==2025.3.0
ftfy==6.3.1 ftfy==6.3.1
# via open-clip-torch # via open-clip-torch
genai-perf==0.0.16 genai-perf==0.0.16
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
genson==1.3.0 genson==1.3.0
# via datamodel-code-generator # via datamodel-code-generator
geopandas==1.1.3 geopandas==1.1.3
...@@ -325,7 +325,7 @@ geopandas==1.1.3 ...@@ -325,7 +325,7 @@ geopandas==1.1.3
gguf==0.18.0 gguf==0.18.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
gitdb==4.0.12 gitdb==4.0.12
# via gitpython # via gitpython
gitpython==3.1.46 gitpython==3.1.46
...@@ -357,7 +357,7 @@ googleapis-common-protos==1.73.0 ...@@ -357,7 +357,7 @@ googleapis-common-protos==1.73.0
# opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-grpc
# opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-otlp-proto-http
gpt-oss==0.0.8 gpt-oss==0.0.8
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
graphql-core==3.2.8 graphql-core==3.2.8
# via hypothesis-graphql # via hypothesis-graphql
greenlet==3.3.2 greenlet==3.3.2
...@@ -365,7 +365,7 @@ greenlet==3.3.2 ...@@ -365,7 +365,7 @@ greenlet==3.3.2
grpcio==1.78.0 grpcio==1.78.0
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# grpcio-reflection # grpcio-reflection
# opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-grpc
# ray # ray
...@@ -373,7 +373,7 @@ grpcio==1.78.0 ...@@ -373,7 +373,7 @@ grpcio==1.78.0
grpcio-reflection==1.78.0 grpcio-reflection==1.78.0
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
h11==0.16.0 h11==0.16.0
# via # via
# httpcore # httpcore
...@@ -398,7 +398,7 @@ httptools==0.7.1 ...@@ -398,7 +398,7 @@ httptools==0.7.1
# via uvicorn # via uvicorn
httpx==0.27.2 httpx==0.27.2
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# anthropic # anthropic
# diffusers # diffusers
# fastapi # fastapi
...@@ -412,7 +412,7 @@ httpx-sse==0.4.3 ...@@ -412,7 +412,7 @@ httpx-sse==0.4.3
# via mcp # via mcp
huggingface-hub==0.36.2 huggingface-hub==0.36.2
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# accelerate # accelerate
# datasets # datasets
# diffusers # diffusers
...@@ -452,9 +452,9 @@ idna==3.11 ...@@ -452,9 +452,9 @@ idna==3.11
# requests # requests
# yarl # yarl
ijson==3.5.0 ijson==3.5.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
imagehash==4.3.2 imagehash==4.3.2
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
imageio==2.37.3 imageio==2.37.3
# via scikit-image # via scikit-image
importlib-metadata==8.7.1 importlib-metadata==8.7.1
...@@ -468,7 +468,7 @@ inflect==7.5.0 ...@@ -468,7 +468,7 @@ inflect==7.5.0
iniconfig==2.3.0 iniconfig==2.3.0
# via pytest # via pytest
instanttensor==0.1.6 instanttensor==0.1.6
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
interegular==0.3.3 interegular==0.3.3
# via lm-format-enforcer # via lm-format-enforcer
isodate==0.7.2 isodate==0.7.2
...@@ -489,7 +489,7 @@ jiter==0.13.0 ...@@ -489,7 +489,7 @@ jiter==0.13.0
# anthropic # anthropic
# openai # openai
jiwer==4.0.0 jiwer==4.0.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
jmespath==1.1.0 jmespath==1.1.0
# via # via
# boto3 # boto3
...@@ -522,7 +522,7 @@ jsonschema-specifications==2025.9.1 ...@@ -522,7 +522,7 @@ jsonschema-specifications==2025.9.1
junit-xml==1.9 junit-xml==1.9
# via schemathesis # via schemathesis
kaldi-native-fbank==1.22.3 kaldi-native-fbank==1.22.3
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
kaleido==1.0.0 kaleido==1.0.0
# via genai-perf # via genai-perf
kiwisolver==1.5.0 kiwisolver==1.5.0
...@@ -534,7 +534,7 @@ kornia-rs==0.1.10 ...@@ -534,7 +534,7 @@ kornia-rs==0.1.10
lark==1.2.2 lark==1.2.2
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
lazy-loader==0.4 lazy-loader==0.4
# via # via
# librosa # librosa
...@@ -542,7 +542,7 @@ lazy-loader==0.4 ...@@ -542,7 +542,7 @@ lazy-loader==0.4
libnacl==2.1.0 libnacl==2.1.0
# via tensorizer # via tensorizer
librosa==0.10.2.post1 librosa==0.10.2.post1
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
lightly==1.5.22 lightly==1.5.22
# via # via
# terratorch # terratorch
...@@ -561,15 +561,15 @@ lightning-utilities==0.15.3 ...@@ -561,15 +561,15 @@ lightning-utilities==0.15.3
llguidance==1.3.0 llguidance==1.3.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
llvmlite==0.44.0 llvmlite==0.44.0
# via numba # via numba
lm-eval==0.4.11 lm-eval==0.4.11
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
lm-format-enforcer==0.11.3 lm-format-enforcer==0.11.3
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
logistro==2.0.1 logistro==2.0.1
# via # via
# choreographer # choreographer
...@@ -594,7 +594,7 @@ markupsafe==3.0.3 ...@@ -594,7 +594,7 @@ markupsafe==3.0.3
# werkzeug # werkzeug
matplotlib==3.10.8 matplotlib==3.10.8
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# lightning # lightning
# torchgeo # torchgeo
mbstrdecoder==1.1.4 mbstrdecoder==1.1.4
...@@ -603,18 +603,18 @@ mbstrdecoder==1.1.4 ...@@ -603,18 +603,18 @@ mbstrdecoder==1.1.4
# pytablewriter # pytablewriter
# typepy # typepy
mcp==1.27.0 mcp==1.27.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
mdurl==0.1.2 mdurl==0.1.2
# via markdown-it-py # via markdown-it-py
mistral-common==1.11.0 mistral-common==1.11.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
model-hosting-container-standards==0.1.14 model-hosting-container-standards==0.1.14
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
more-itertools==10.8.0 more-itertools==10.8.0
# via # via
# inflect # inflect
...@@ -632,16 +632,16 @@ msgpack==1.1.2 ...@@ -632,16 +632,16 @@ msgpack==1.1.2
# librosa # librosa
# ray # ray
msgspec==0.20.0 msgspec==0.20.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
mteb==2.11.5 mteb==2.11.5
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
multidict==6.7.1 multidict==6.7.1
# via # via
# aiohttp # aiohttp
# yarl # yarl
multiprocess==0.70.16 multiprocess==0.70.16
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# datasets # datasets
# evaluate # evaluate
mypy-extensions==1.1.0 mypy-extensions==1.1.0
...@@ -653,23 +653,23 @@ networkx==3.6.1 ...@@ -653,23 +653,23 @@ networkx==3.6.1
# scikit-image # scikit-image
# torch # torch
ninja==1.13.0 ninja==1.13.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
nltk==3.9.3 nltk==3.9.3
# via rouge-score # via rouge-score
num2words==0.5.14 num2words==0.5.14
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
numba==0.61.2 numba==0.61.2
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# librosa # librosa
# resampy # resampy
numkong==7.1.1 numkong==7.1.1
# via albucore # via albucore
numpy==2.2.6 numpy==2.2.6
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# accelerate # accelerate
# albucore # albucore
# albumentations # albumentations
...@@ -741,15 +741,15 @@ omegaconf==2.3.0 ...@@ -741,15 +741,15 @@ omegaconf==2.3.0
# hydra-core # hydra-core
# lightning # lightning
open-clip-torch==2.32.0 open-clip-torch==2.32.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
openai==2.30.0 openai==2.30.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
openai-harmony==0.0.8 openai-harmony==0.0.8
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# gpt-oss # gpt-oss
opencensus==0.11.4 opencensus==0.11.4
# via ray # via ray
...@@ -758,16 +758,16 @@ opencensus-context==0.1.3 ...@@ -758,16 +758,16 @@ opencensus-context==0.1.3
opencv-python-headless==4.13.0.92 opencv-python-headless==4.13.0.92
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# albumentations # albumentations
# mistral-common # mistral-common
openpyxl==3.1.5 openpyxl==3.1.5
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
opentelemetry-api==1.40.0 opentelemetry-api==1.40.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-grpc
# opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-otlp-proto-http
# opentelemetry-exporter-prometheus # opentelemetry-exporter-prometheus
...@@ -776,7 +776,7 @@ opentelemetry-api==1.40.0 ...@@ -776,7 +776,7 @@ opentelemetry-api==1.40.0
opentelemetry-exporter-otlp==1.40.0 opentelemetry-exporter-otlp==1.40.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
opentelemetry-exporter-otlp-proto-common==1.40.0 opentelemetry-exporter-otlp-proto-common==1.40.0
# via # via
# opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-grpc
...@@ -796,7 +796,7 @@ opentelemetry-proto==1.40.0 ...@@ -796,7 +796,7 @@ opentelemetry-proto==1.40.0
opentelemetry-sdk==1.40.0 opentelemetry-sdk==1.40.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-grpc
# opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-otlp-proto-http
# opentelemetry-exporter-prometheus # opentelemetry-exporter-prometheus
...@@ -809,7 +809,7 @@ opentelemetry-semantic-conventions==0.61b0 ...@@ -809,7 +809,7 @@ opentelemetry-semantic-conventions==0.61b0
opentelemetry-semantic-conventions-ai==0.5.1 opentelemetry-semantic-conventions-ai==0.5.1
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
optuna==3.6.1 optuna==3.6.1
# via genai-perf # via genai-perf
orjson==3.11.7 orjson==3.11.7
...@@ -819,7 +819,7 @@ orjson==3.11.7 ...@@ -819,7 +819,7 @@ orjson==3.11.7
outlines-core==0.2.11 outlines-core==0.2.11
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
packaging==26.0 packaging==26.0
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
...@@ -870,7 +870,7 @@ pandas==3.0.1 ...@@ -870,7 +870,7 @@ pandas==3.0.1
# torchgeo # torchgeo
# xarray # xarray
partial-json-parser==0.2.1.1.post7 partial-json-parser==0.2.1.1.post7
# via -r requirements/common.txt # via -r requirements/test/../common.txt
pathspec==1.0.4 pathspec==1.0.4
# via black # via black
pathvalidate==3.3.1 pathvalidate==3.3.1
...@@ -878,14 +878,14 @@ pathvalidate==3.3.1 ...@@ -878,14 +878,14 @@ pathvalidate==3.3.1
patsy==1.0.2 patsy==1.0.2
# via statsmodels # via statsmodels
peft==0.18.1 peft==0.18.1
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
perceptron==0.1.4 perceptron==0.1.4
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
perf-analyzer==0.1.0 perf-analyzer==0.1.0
# via genai-perf # via genai-perf
pillow==12.1.1 pillow==12.1.1
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# diffusers # diffusers
# genai-perf # genai-perf
# imagehash # imagehash
...@@ -908,7 +908,7 @@ platformdirs==4.3.6 ...@@ -908,7 +908,7 @@ platformdirs==4.3.6
# wandb # wandb
plotly==6.6.0 plotly==6.6.0
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# genai-perf # genai-perf
pluggy==1.6.0 pluggy==1.6.0
# via # via
...@@ -923,18 +923,18 @@ pooch==1.8.2 ...@@ -923,18 +923,18 @@ pooch==1.8.2
portalocker==3.2.0 portalocker==3.2.0
# via sacrebleu # via sacrebleu
pqdm==0.2.0 pqdm==0.2.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
prometheus-client==0.24.1 prometheus-client==0.24.1
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# opentelemetry-exporter-prometheus # opentelemetry-exporter-prometheus
# prometheus-fastapi-instrumentator # prometheus-fastapi-instrumentator
# ray # ray
prometheus-fastapi-instrumentator==7.1.0 prometheus-fastapi-instrumentator==7.1.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
propcache==0.4.1 propcache==0.4.1
# via # via
# aiohttp # aiohttp
...@@ -944,7 +944,7 @@ proto-plus==1.27.1 ...@@ -944,7 +944,7 @@ proto-plus==1.27.1
protobuf==6.33.6 protobuf==6.33.6
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# google-api-core # google-api-core
# googleapis-common-protos # googleapis-common-protos
# grpcio-reflection # grpcio-reflection
...@@ -957,14 +957,14 @@ protobuf==6.33.6 ...@@ -957,14 +957,14 @@ protobuf==6.33.6
# wandb # wandb
psutil==7.2.2 psutil==7.2.2
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# accelerate # accelerate
# peft # peft
# tensorizer # tensorizer
py==1.11.0 py==1.11.0
# via pytest-forked # via pytest-forked
py-cpuinfo==9.0.0 py-cpuinfo==9.0.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
py-spy==0.4.1 py-spy==0.4.1
# via ray # via ray
pyarrow==23.0.1 pyarrow==23.0.1
...@@ -978,7 +978,7 @@ pyasn1==0.6.3 ...@@ -978,7 +978,7 @@ pyasn1==0.6.3
pyasn1-modules==0.4.2 pyasn1-modules==0.4.2
# via google-auth # via google-auth
pybase64==1.4.3 pybase64==1.4.3
# via -r requirements/common.txt # via -r requirements/test/../common.txt
pycocotools==2.0.11 pycocotools==2.0.11
# via terratorch # via terratorch
pycountry==26.2.16 pycountry==26.2.16
...@@ -990,8 +990,8 @@ pycryptodomex==3.23.0 ...@@ -990,8 +990,8 @@ pycryptodomex==3.23.0
pydantic==2.12.5 pydantic==2.12.5
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# albumentations # albumentations
# anthropic # anthropic
# compressed-tensors # compressed-tensors
...@@ -1047,7 +1047,7 @@ pytablewriter==1.2.1 ...@@ -1047,7 +1047,7 @@ pytablewriter==1.2.1
# via lm-eval # via lm-eval
pytest==8.3.5 pytest==8.3.5
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# buildkite-test-collector # buildkite-test-collector
# genai-perf # genai-perf
# pytest-asyncio # pytest-asyncio
...@@ -1060,21 +1060,21 @@ pytest==8.3.5 ...@@ -1060,21 +1060,21 @@ pytest==8.3.5
# pytest-timeout # pytest-timeout
# schemathesis # schemathesis
pytest-asyncio==0.24.0 pytest-asyncio==0.24.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
pytest-cov==6.3.0 pytest-cov==6.3.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
pytest-forked==1.6.0 pytest-forked==1.6.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
pytest-mock==3.15.1 pytest-mock==3.15.1
# via genai-perf # via genai-perf
pytest-rerunfailures==14.0 pytest-rerunfailures==14.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
pytest-shard==0.1.2 pytest-shard==0.1.2
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
pytest-subtests==0.14.2 pytest-subtests==0.14.2
# via schemathesis # via schemathesis
pytest-timeout==2.3.1 pytest-timeout==2.3.1
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
python-box==7.4.1 python-box==7.4.1
# via terratorch # via terratorch
python-dateutil==2.9.0.post0 python-dateutil==2.9.0.post0
...@@ -1092,7 +1092,7 @@ python-dotenv==1.2.2 ...@@ -1092,7 +1092,7 @@ python-dotenv==1.2.2
# pydantic-settings # pydantic-settings
# uvicorn # uvicorn
python-json-logger==4.1.0 python-json-logger==4.1.0
# via -r requirements/common.txt # via -r requirements/test/../common.txt
python-multipart==0.0.22 python-multipart==0.0.22
# via # via
# fastapi # fastapi
...@@ -1113,7 +1113,7 @@ pywavelets==1.9.0 ...@@ -1113,7 +1113,7 @@ pywavelets==1.9.0
# via imagehash # via imagehash
pyyaml==6.0.3 pyyaml==6.0.3
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# accelerate # accelerate
# albumentations # albumentations
# datamodel-code-generator # datamodel-code-generator
...@@ -1139,10 +1139,10 @@ pyyaml==6.0.3 ...@@ -1139,10 +1139,10 @@ pyyaml==6.0.3
pyzmq==27.1.0 pyzmq==27.1.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
rapidfuzz==3.12.1 rapidfuzz==3.12.1
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# jiwer # jiwer
rasterio==1.5.0 rasterio==1.5.0
# via # via
...@@ -1150,7 +1150,7 @@ rasterio==1.5.0 ...@@ -1150,7 +1150,7 @@ rasterio==1.5.0
# terratorch # terratorch
# torchgeo # torchgeo
ray==2.54.0 ray==2.54.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
redis==7.3.0 redis==7.3.0
# via tensorizer # via tensorizer
referencing==0.37.0 referencing==0.37.0
...@@ -1159,7 +1159,7 @@ referencing==0.37.0 ...@@ -1159,7 +1159,7 @@ referencing==0.37.0
# jsonschema-specifications # jsonschema-specifications
regex==2026.2.28 regex==2026.2.28
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# diffusers # diffusers
# nltk # nltk
# open-clip-torch # open-clip-torch
...@@ -1169,7 +1169,7 @@ regex==2026.2.28 ...@@ -1169,7 +1169,7 @@ regex==2026.2.28
requests==2.32.5 requests==2.32.5
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# azure-core # azure-core
# buildkite-test-collector # buildkite-test-collector
# datasets # datasets
...@@ -1197,7 +1197,7 @@ requests==2.32.5 ...@@ -1197,7 +1197,7 @@ requests==2.32.5
# transformers # transformers
# wandb # wandb
resampy==0.4.3 resampy==0.4.3
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
responses==0.26.0 responses==0.26.0
# via genai-perf # via genai-perf
rfc3339-validator==0.1.4 rfc3339-validator==0.1.4
...@@ -1232,7 +1232,7 @@ rtree==1.4.1 ...@@ -1232,7 +1232,7 @@ rtree==1.4.1
runai-model-streamer==0.15.7 runai-model-streamer==0.15.7
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
runai-model-streamer-azure==0.15.7 runai-model-streamer-azure==0.15.7
# via runai-model-streamer # via runai-model-streamer
runai-model-streamer-gcs==0.15.7 runai-model-streamer-gcs==0.15.7
...@@ -1253,7 +1253,7 @@ safetensors==0.7.0 ...@@ -1253,7 +1253,7 @@ safetensors==0.7.0
# timm # timm
# transformers # transformers
schemathesis==3.39.15 schemathesis==3.39.15
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
scikit-image==0.26.0 scikit-image==0.26.0
# via # via
# albumentations # albumentations
...@@ -1281,26 +1281,26 @@ scipy==1.17.1 ...@@ -1281,26 +1281,26 @@ scipy==1.17.1
# vocos # vocos
segmentation-models-pytorch==0.5.0 segmentation-models-pytorch==0.5.0
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# terratorch # terratorch
# torchgeo # torchgeo
sentence-transformers==5.3.0 sentence-transformers==5.3.0
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# mteb # mteb
sentencepiece==0.2.1 sentencepiece==0.2.1
# via -r requirements/common.txt # via -r requirements/test/../common.txt
sentry-sdk==2.55.0 sentry-sdk==2.55.0
# via # via
# fastapi-cloud-cli # fastapi-cloud-cli
# wandb # wandb
setproctitle==1.3.7 setproctitle==1.3.7
# via -r requirements/common.txt # via -r requirements/test/../common.txt
setuptools==79.0.1 setuptools==79.0.1
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# model-hosting-container-standards # model-hosting-container-standards
# pytablewriter # pytablewriter
# tensorboard # tensorboard
...@@ -1318,7 +1318,7 @@ simplejson==3.20.2 ...@@ -1318,7 +1318,7 @@ simplejson==3.20.2
six==1.17.0 six==1.17.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# junit-xml # junit-xml
# lightly # lightly
# opencensus # opencensus
...@@ -1338,7 +1338,7 @@ sortedcontainers==2.4.0 ...@@ -1338,7 +1338,7 @@ sortedcontainers==2.4.0
# via hypothesis # via hypothesis
soundfile==0.13.1 soundfile==0.13.1
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# genai-perf # genai-perf
# librosa # librosa
# mistral-common # mistral-common
...@@ -1384,7 +1384,7 @@ tabulate==0.10.0 ...@@ -1384,7 +1384,7 @@ tabulate==0.10.0
tacoreader==0.5.6 tacoreader==0.5.6
# via terratorch # via terratorch
tblib==3.1.0 tblib==3.1.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
tcolorpy==0.1.7 tcolorpy==0.1.7
# via pytablewriter # via pytablewriter
tenacity==9.1.4 tenacity==9.1.4
...@@ -1400,13 +1400,13 @@ tensorboardx==2.6.4 ...@@ -1400,13 +1400,13 @@ tensorboardx==2.6.4
tensorizer==2.10.1 tensorizer==2.10.1
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
termcolor==3.3.0 termcolor==3.3.0
# via # via
# gpt-oss # gpt-oss
# terratorch # terratorch
terratorch==1.2.2 terratorch==1.2.2
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
threadpoolctl==3.6.0 threadpoolctl==3.6.0
# via scikit-learn # via scikit-learn
tifffile==2026.3.3 tifffile==2026.3.3
...@@ -1416,14 +1416,14 @@ tifffile==2026.3.3 ...@@ -1416,14 +1416,14 @@ tifffile==2026.3.3
tiktoken==0.12.0 tiktoken==0.12.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# gpt-oss # gpt-oss
# lm-eval # lm-eval
# mistral-common # mistral-common
timm==1.0.17 timm==1.0.17
# via # via
# -c requirements/rocm.txt # -c requirements/rocm.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# open-clip-torch # open-clip-torch
# segmentation-models-pytorch # segmentation-models-pytorch
# terratorch # terratorch
...@@ -1431,8 +1431,8 @@ timm==1.0.17 ...@@ -1431,8 +1431,8 @@ timm==1.0.17
tokenizers==0.22.0 tokenizers==0.22.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# transformers # transformers
tomli==2.4.0 tomli==2.4.0
# via schemathesis # via schemathesis
...@@ -1440,7 +1440,7 @@ tomli-w==1.2.0 ...@@ -1440,7 +1440,7 @@ tomli-w==1.2.0
# via schemathesis # via schemathesis
torchgeo==0.7.0 torchgeo==0.7.0
# via # via
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# terratorch # terratorch
torchmetrics==1.9.0 torchmetrics==1.9.0
# via # via
...@@ -1450,7 +1450,7 @@ torchmetrics==1.9.0 ...@@ -1450,7 +1450,7 @@ torchmetrics==1.9.0
# torchgeo # torchgeo
tqdm==4.67.3 tqdm==4.67.3
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# datasets # datasets
# evaluate # evaluate
# gguf # gguf
...@@ -1474,8 +1474,8 @@ tqdm==4.67.3 ...@@ -1474,8 +1474,8 @@ tqdm==4.67.3
transformers==4.57.5 transformers==4.57.5
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# -r requirements/rocm-test.in # -r requirements/test/rocm.in
# compressed-tensors # compressed-tensors
# genai-perf # genai-perf
# peft # peft
...@@ -1483,9 +1483,9 @@ transformers==4.57.5 ...@@ -1483,9 +1483,9 @@ transformers==4.57.5
# transformers-stream-generator # transformers-stream-generator
# xgrammar # xgrammar
transformers-stream-generator==0.0.5 transformers-stream-generator==0.0.5
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
tritonclient==2.66.0 tritonclient==2.66.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
typeguard==4.5.1 typeguard==4.5.1
# via inflect # via inflect
typepy==1.3.4 typepy==1.3.4
...@@ -1504,7 +1504,7 @@ typeshed-client==2.9.0 ...@@ -1504,7 +1504,7 @@ typeshed-client==2.9.0
typing-extensions==4.15.0 typing-extensions==4.15.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
# aiosignal # aiosignal
# albumentations # albumentations
# alembic # alembic
...@@ -1577,16 +1577,16 @@ uvicorn==0.42.0 ...@@ -1577,16 +1577,16 @@ uvicorn==0.42.0
uvloop==0.22.1 uvloop==0.22.1
# via uvicorn # via uvicorn
vector-quantize-pytorch==1.28.0 vector-quantize-pytorch==1.28.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
virtualenv==21.2.0 virtualenv==21.2.0
# via ray # via ray
vocos==0.1.0 vocos==0.1.0
# via -r requirements/rocm-test.in # via -r requirements/test/rocm.in
wandb==0.25.1 wandb==0.25.1
# via terratorch # via terratorch
watchfiles==1.1.1 watchfiles==1.1.1
# via # via
# -r requirements/common.txt # -r requirements/test/../common.txt
# uvicorn # uvicorn
wcwidth==0.6.0 wcwidth==0.6.0
# via ftfy # via ftfy
...@@ -1607,7 +1607,7 @@ xarray==2026.2.0 ...@@ -1607,7 +1607,7 @@ xarray==2026.2.0
xgrammar==0.1.33 xgrammar==0.1.33
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/common.txt # -r requirements/test/../common.txt
xxhash==3.6.0 xxhash==3.6.0
# via # via
# datasets # datasets
......
requirements/xpu-test.txt → requirements/test/xpu.txt
View file @ fc645f1a
# This file was autogenerated by uv via the following command: # This file was autogenerated by uv via the following command:
# uv pip compile requirements/xpu-test.in -o requirements/xpu-test.txt -c requirements/xpu.txt --python-version 3.12 --index-strategy unsafe-best-match # uv pip compile requirements/test/xpu.in -c requirements/xpu.txt -o requirements/test/xpu.txt --index-strategy unsafe-best-match --torch-backend xpu --python-platform x86_64-manylinux_2_39 --python-version 3.12
absl-py==2.4.0 absl-py==2.4.0
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# rouge-score # rouge-score
accelerate==1.13.0 accelerate==1.13.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
aiohappyeyeballs==2.6.1 aiohappyeyeballs==2.6.1
# via aiohttp # via aiohttp
aiohttp==3.13.4 aiohttp==3.13.4
...@@ -17,7 +17,7 @@ aiohttp==3.13.4 ...@@ -17,7 +17,7 @@ aiohttp==3.13.4
aiosignal==1.4.0 aiosignal==1.4.0
# via aiohttp # via aiohttp
albumentations==1.4.6 albumentations==1.4.6
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
annotated-doc==0.0.4 annotated-doc==0.0.4
# via fastapi # via fastapi
annotated-types==0.7.0 annotated-types==0.7.0
...@@ -27,7 +27,7 @@ anyio==4.13.0 ...@@ -27,7 +27,7 @@ anyio==4.13.0
# httpx # httpx
# starlette # starlette
arctic-inference==0.1.1 arctic-inference==0.1.1
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
attrs==26.1.0 attrs==26.1.0
# via # via
# aiohttp # aiohttp
...@@ -36,13 +36,13 @@ attrs==26.1.0 ...@@ -36,13 +36,13 @@ attrs==26.1.0
# referencing # referencing
audioread==3.0.1 audioread==3.0.1
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# librosa # librosa
blobfile==3.0.0 blobfile==3.0.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
bm25s==0.2.13 bm25s==0.2.13
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# mteb # mteb
bounded-pool-executor==0.0.3 bounded-pool-executor==0.0.3
# via pqdm # via pqdm
...@@ -90,7 +90,7 @@ docker==7.1.0 ...@@ -90,7 +90,7 @@ docker==7.1.0
# via gpt-oss # via gpt-oss
docopt==0.6.2 docopt==0.6.2
# via num2words # via num2words
dpcpp-cpp-rt==2025.3.1 dpcpp-cpp-rt==2025.3.2
# via # via
# onemkl-sycl-blas # onemkl-sycl-blas
# onemkl-sycl-dft # onemkl-sycl-dft
...@@ -124,7 +124,7 @@ fsspec==2026.2.0 ...@@ -124,7 +124,7 @@ fsspec==2026.2.0
# huggingface-hub # huggingface-hub
# torch # torch
gpt-oss==0.0.8 gpt-oss==0.0.8
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
graphql-core==3.2.8 graphql-core==3.2.8
# via hypothesis-graphql # via hypothesis-graphql
h11==0.16.0 h11==0.16.0
...@@ -134,7 +134,7 @@ h11==0.16.0 ...@@ -134,7 +134,7 @@ h11==0.16.0
harfile==0.4.0 harfile==0.4.0
# via schemathesis # via schemathesis
hf-transfer==0.1.9 hf-transfer==0.1.9
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
hf-xet==1.4.2 hf-xet==1.4.2
# via huggingface-hub # via huggingface-hub
html2text==2025.4.15 html2text==2025.4.15
...@@ -171,27 +171,27 @@ idna==3.11 ...@@ -171,27 +171,27 @@ idna==3.11
# yarl # yarl
imageio==2.37.3 imageio==2.37.3
# via scikit-image # via scikit-image
impi-rt==2021.17.0 impi-rt==2021.17.2
# via # via
# oneccl # oneccl
# torch # torch
iniconfig==2.3.0 iniconfig==2.3.0
# via pytest # via pytest
intel-cmplr-lib-rt==2025.3.1 intel-cmplr-lib-rt==2025.3.2
# via # via
# intel-sycl-rt # intel-sycl-rt
# torch # torch
intel-cmplr-lib-ur==2025.3.1 intel-cmplr-lib-ur==2025.3.2
# via # via
# intel-openmp # intel-openmp
# intel-sycl-rt # intel-sycl-rt
# torch # torch
intel-cmplr-lic-rt==2025.3.1 intel-cmplr-lic-rt==2025.3.2
# via # via
# intel-opencl-rt # intel-opencl-rt
# intel-sycl-rt # intel-sycl-rt
# torch # torch
intel-opencl-rt==2025.3.1 intel-opencl-rt==2025.3.2
# via # via
# dpcpp-cpp-rt # dpcpp-cpp-rt
# onemkl-sycl-blas # onemkl-sycl-blas
...@@ -200,14 +200,14 @@ intel-opencl-rt==2025.3.1 ...@@ -200,14 +200,14 @@ intel-opencl-rt==2025.3.1
# onemkl-sycl-rng # onemkl-sycl-rng
# onemkl-sycl-sparse # onemkl-sycl-sparse
# torch # torch
intel-openmp==2025.3.1 intel-openmp==2025.3.2
# via # via
# dpcpp-cpp-rt # dpcpp-cpp-rt
# mkl # mkl
# torch # torch
intel-pti==0.15.0 intel-pti==0.16.0
# via torch # via torch
intel-sycl-rt==2025.3.1 intel-sycl-rt==2025.3.2
# via # via
# dpcpp-cpp-rt # dpcpp-cpp-rt
# oneccl # oneccl
...@@ -218,7 +218,7 @@ jinja2==3.1.6 ...@@ -218,7 +218,7 @@ jinja2==3.1.6
# lm-eval # lm-eval
# torch # torch
jiwer==4.0.0 jiwer==4.0.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
joblib==1.5.3 joblib==1.5.3
# via # via
# librosa # librosa
...@@ -242,11 +242,11 @@ lazy-loader==0.5 ...@@ -242,11 +242,11 @@ lazy-loader==0.5
# librosa # librosa
# scikit-image # scikit-image
librosa==0.10.2.post1 librosa==0.10.2.post1
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
llvmlite==0.44.0 llvmlite==0.44.0
# via numba # via numba
lm-eval==0.4.11 lm-eval==0.4.11
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
lxml==6.0.2 lxml==6.0.2
# via # via
# blobfile # blobfile
...@@ -265,11 +265,11 @@ mbstrdecoder==1.1.4 ...@@ -265,11 +265,11 @@ mbstrdecoder==1.1.4
# typepy # typepy
mdurl==0.1.2 mdurl==0.1.2
# via markdown-it-py # via markdown-it-py
mistral-common==1.10.0 mistral-common==1.11.0
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
mkl==2025.3.0 mkl==2025.3.1
# via # via
# onemkl-sycl-blas # onemkl-sycl-blas
# onemkl-sycl-dft # onemkl-sycl-dft
...@@ -278,7 +278,7 @@ mkl==2025.3.0 ...@@ -278,7 +278,7 @@ mkl==2025.3.0
# onemkl-sycl-sparse # onemkl-sycl-sparse
# torch # torch
modelscope==1.35.3 modelscope==1.35.3
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
more-itertools==10.8.0 more-itertools==10.8.0
# via lm-eval # via lm-eval
mpmath==1.3.0 mpmath==1.3.0
...@@ -286,7 +286,7 @@ mpmath==1.3.0 ...@@ -286,7 +286,7 @@ mpmath==1.3.0
msgpack==1.1.2 msgpack==1.1.2
# via librosa # via librosa
mteb==2.12.7 mteb==2.12.7
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
multidict==6.7.1 multidict==6.7.1
# via # via
# aiohttp # aiohttp
...@@ -302,7 +302,7 @@ networkx==3.6.1 ...@@ -302,7 +302,7 @@ networkx==3.6.1
nltk==3.9.4 nltk==3.9.4
# via rouge-score # via rouge-score
num2words==0.5.14 num2words==0.5.14
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
numba==0.61.2 numba==0.61.2
# via # via
# -c requirements/xpu.txt # -c requirements/xpu.txt
...@@ -334,28 +334,28 @@ numpy==2.2.6 ...@@ -334,28 +334,28 @@ numpy==2.2.6
# tifffile # tifffile
# torchvision # torchvision
# transformers # transformers
oneccl==2021.17.1 oneccl==2021.17.2
# via # via
# oneccl-devel # oneccl-devel
# torch # torch
oneccl-devel==2021.17.1 oneccl-devel==2021.17.2
# via torch # via torch
onemkl-license==2025.3.0 onemkl-license==2025.3.1
# via # via
# mkl # mkl
# torch # torch
onemkl-sycl-blas==2025.3.0 onemkl-sycl-blas==2025.3.1
# via # via
# onemkl-sycl-lapack # onemkl-sycl-lapack
# onemkl-sycl-sparse # onemkl-sycl-sparse
# torch # torch
onemkl-sycl-dft==2025.3.0 onemkl-sycl-dft==2025.3.1
# via torch # via torch
onemkl-sycl-lapack==2025.3.0 onemkl-sycl-lapack==2025.3.1
# via torch # via torch
onemkl-sycl-rng==2025.3.0 onemkl-sycl-rng==2025.3.1
# via torch # via torch
onemkl-sycl-sparse==2025.3.0 onemkl-sycl-sparse==2025.3.1
# via torch # via torch
openai-harmony==0.0.8 openai-harmony==0.0.8
# via # via
...@@ -405,12 +405,12 @@ polars-runtime-32==1.39.3 ...@@ -405,12 +405,12 @@ polars-runtime-32==1.39.3
# via polars # via polars
pooch==1.8.2 pooch==1.8.2
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# librosa # librosa
portalocker==3.2.0 portalocker==3.2.0
# via sacrebleu # via sacrebleu
pqdm==0.2.0 pqdm==0.2.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
propcache==0.4.1 propcache==0.4.1
# via # via
# aiohttp # aiohttp
...@@ -451,13 +451,13 @@ pyrate-limiter==4.1.0 ...@@ -451,13 +451,13 @@ pyrate-limiter==4.1.0
# via schemathesis # via schemathesis
pystemmer==3.0.0 pystemmer==3.0.0
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# mteb # mteb
pytablewriter==1.2.1 pytablewriter==1.2.1
# via lm-eval # via lm-eval
pytest==9.0.2 pytest==9.0.2
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# pytest-asyncio # pytest-asyncio
# pytest-cov # pytest-cov
# pytest-forked # pytest-forked
...@@ -466,17 +466,17 @@ pytest==9.0.2 ...@@ -466,17 +466,17 @@ pytest==9.0.2
# pytest-timeout # pytest-timeout
# schemathesis # schemathesis
pytest-asyncio==1.3.0 pytest-asyncio==1.3.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
pytest-cov==6.3.0 pytest-cov==6.3.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
pytest-forked==1.6.0 pytest-forked==1.6.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
pytest-rerunfailures==14.0 pytest-rerunfailures==14.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
pytest-shard==0.1.2 pytest-shard==0.1.2
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
pytest-timeout==2.3.1 pytest-timeout==2.3.1
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
python-dateutil==2.9.0.post0 python-dateutil==2.9.0.post0
# via # via
# pandas # pandas
...@@ -496,7 +496,7 @@ pyyaml==6.0.3 ...@@ -496,7 +496,7 @@ pyyaml==6.0.3
# transformers # transformers
rapidfuzz==3.12.1 rapidfuzz==3.12.1
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# jiwer # jiwer
referencing==0.37.0 referencing==0.37.0
# via # via
...@@ -543,7 +543,7 @@ safetensors==0.7.0 ...@@ -543,7 +543,7 @@ safetensors==0.7.0
# timm # timm
# transformers # transformers
schemathesis==4.14.2 schemathesis==4.14.2
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
scikit-image==0.26.0 scikit-image==0.26.0
# via albumentations # via albumentations
scikit-learn==1.8.0 scikit-learn==1.8.0
...@@ -582,12 +582,12 @@ sortedcontainers==2.4.0 ...@@ -582,12 +582,12 @@ sortedcontainers==2.4.0
# via hypothesis # via hypothesis
soundfile==0.13.1 soundfile==0.13.1
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# librosa # librosa
# mistral-common # mistral-common
soxr==0.5.0.post1 soxr==0.5.0.post1
# via # via
# -r requirements/xpu-test.in # -r requirements/test/xpu.in
# librosa # librosa
# mistral-common # mistral-common
sqlitedict==2.1.0 sqlitedict==2.1.0
...@@ -606,13 +606,13 @@ tabledata==1.3.4 ...@@ -606,13 +606,13 @@ tabledata==1.3.4
# via pytablewriter # via pytablewriter
tabulate==0.10.0 tabulate==0.10.0
# via sacrebleu # via sacrebleu
tbb==2022.3.0 tbb==2022.3.1
# via # via
# intel-opencl-rt # intel-opencl-rt
# mkl # mkl
# torch # torch
tblib==3.1.0 tblib==3.1.0
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
tcmlib==1.4.1 tcmlib==1.4.1
# via # via
# tbb # tbb
...@@ -638,12 +638,12 @@ tiktoken==0.12.0 ...@@ -638,12 +638,12 @@ tiktoken==0.12.0
# lm-eval # lm-eval
# mistral-common # mistral-common
timm==1.0.17 timm==1.0.17
# via -r requirements/xpu-test.in # via -r requirements/test/xpu.in
tokenizers==0.22.2 tokenizers==0.22.2
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# transformers # transformers
torch==2.10.0+xpu torch==2.11.0+xpu
# via # via
# -c requirements/xpu.txt # -c requirements/xpu.txt
# accelerate # accelerate
...@@ -651,7 +651,7 @@ torch==2.10.0+xpu ...@@ -651,7 +651,7 @@ torch==2.10.0+xpu
# sentence-transformers # sentence-transformers
# timm # timm
# torchvision # torchvision
torchvision==0.25.0+xpu torchvision==0.26.0+xpu
# via timm # via timm
tqdm==4.67.3 tqdm==4.67.3
# via # via
...@@ -669,7 +669,7 @@ transformers==4.57.6 ...@@ -669,7 +669,7 @@ transformers==4.57.6
# via # via
# -c requirements/common.txt # -c requirements/common.txt
# sentence-transformers # sentence-transformers
triton-xpu==3.6.0 triton-xpu==3.7.0
# via torch # via torch
typepy==1.3.4 typepy==1.3.4
# via # via
...@@ -704,7 +704,7 @@ typing-inspection==0.4.2 ...@@ -704,7 +704,7 @@ typing-inspection==0.4.2
# via # via
# fastapi # fastapi
# pydantic # pydantic
umf==1.0.2 umf==1.0.3
# via # via
# intel-cmplr-lib-ur # intel-cmplr-lib-ur
# torch # torch
......
tests/standalone_tests/pytorch_nightly_dependency.sh
View file @ fc645f1a
...@@ -28,8 +28,8 @@ uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > before.txt ...@@ -28,8 +28,8 @@ uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > before.txt
echo "Before:" echo "Before:"
cat before.txt cat before.txt
echo ">>> Installing requirements/nightly_torch_test.txt" echo ">>> Installing requirements/test/nightly-torch.txt"
uv pip install --quiet -r requirements/nightly_torch_test.txt uv pip install --quiet -r requirements/test/nightly-torch.txt
echo ">>> Capturing torch-related versions after requirements install" echo ">>> Capturing torch-related versions after requirements install"
uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > after.txt uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > after.txt
...@@ -40,7 +40,7 @@ echo ">>> Comparing versions" ...@@ -40,7 +40,7 @@ echo ">>> Comparing versions"
if diff before.txt after.txt; then if diff before.txt after.txt; then
echo "torch version not overridden." echo "torch version not overridden."
else else
echo "torch version overridden by nightly_torch_test.txt, \ echo "torch version overridden by test/nightly-torch.txt, \
if the dependency is not triggered by the pytorch nightly test,\ if the dependency is not triggered by the pytorch nightly test,\
please add the dependency to the list 'white_list' in tools/pre_commit/generate_nightly_torch_test.py" please add the dependency to the list 'white_list' in tools/pre_commit/generate_nightly_torch_test.py"
exit 1 exit 1
......
tools/generate_cmake_presets.py
View file @ fc645f1a
...@@ -128,7 +128,7 @@ def generate_presets(output_path="CMakeUserPresets.json", force_overwrite=False) ...@@ -128,7 +128,7 @@ def generate_presets(output_path="CMakeUserPresets.json", force_overwrite=False)
presets = { presets = {
"version": 6, "version": 6,
# Keep in sync with CMakeLists.txt and requirements/build.txt # Keep in sync with CMakeLists.txt and requirements/build/cuda.txt
"cmakeMinimumRequired": {"major": 3, "minor": 26, "patch": 1}, "cmakeMinimumRequired": {"major": 3, "minor": 26, "patch": 1},
"configurePresets": [configure_preset], "configurePresets": [configure_preset],
"buildPresets": [ "buildPresets": [
......
tools/pre_commit/generate_nightly_torch_test.py
View file @ fc645f1a
...@@ -3,15 +3,15 @@ ...@@ -3,15 +3,15 @@
""" """
Generates specialized requirements files for nightly PyTorch testing. Generates specialized requirements files for nightly PyTorch testing.
This script reads the main test requirements input file (`requirements/test.in`) This script reads the main test requirements input file (`requirements/test/cuda.in`)
and splits its content into two files: and splits its content into two files:
1. `requirements/nightly_torch_test.txt`: Contains dependencies 1. `requirements/test/nightly-torch.txt`: Contains dependencies
except PyTorch-related. except PyTorch-related.
2. `torch_nightly_test.txt`: Contains only PyTorch-related packages. 2. `torch_nightly_test.txt`: Contains only PyTorch-related packages.
""" """
input_file = "requirements/test.in" input_file = "requirements/test/cuda.in"
output_file = "requirements/nightly_torch_test.txt" output_file = "requirements/test/nightly-torch.txt"
# white list of packages that are not compatible with PyTorch nightly directly # white list of packages that are not compatible with PyTorch nightly directly
# with pip install. Please add your package to this list if it is not compatible # with pip install. Please add your package to this list if it is not compatible
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment