diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index 245a86f93beb..6dfa9c8f1ffe 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -8,11 +8,9 @@ aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 # via - # -c requirements/common.txt # aiohttp-cors # datasets # fsspec - # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 @@ -20,19 +18,17 @@ aiohttp-cors==0.8.1 aiosignal==1.4.0 # via aiohttp albumentations==1.4.6 - # via -r requirements/test/cuda.in + # via -r test/cuda.in alembic==1.16.4 # via optuna -annotated-doc==0.0.4 - # via fastapi annotated-types==0.7.0 # via pydantic anyio==4.6.2.post1 # via # httpx # starlette -arctic-inference==0.1.1 - # via -r requirements/test/cuda.in +arctic-inference==0.1.1 ; platform_machine == "x86_64" + # via -r test/cuda.in argcomplete==3.5.1 # via datamodel-code-generator arrow==1.3.0 @@ -47,7 +43,7 @@ attrs==24.2.0 audioread==3.0.1 # via librosa av==16.1.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in azure-core==1.38.2 # via # azure-identity @@ -58,14 +54,14 @@ azure-storage-blob==12.28.0 # via runai-model-streamer-azure backoff==2.2.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # schemathesis bitsandbytes==0.49.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in black==24.10.0 # via datamodel-code-generator blobfile==3.0.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in bm25s==0.2.13 # via mteb boto3==1.35.57 @@ -79,7 +75,7 @@ botocore==1.35.57 bounded-pool-executor==0.0.3 # via pqdm buildkite-test-collector==0.1.9 - # via -r requirements/test/cuda.in + # via -r test/cuda.in cachetools==5.5.2 # via google-auth certifi==2024.8.30 @@ -95,8 +91,6 @@ chardet==5.2.0 # via mbstrdecoder charset-normalizer==3.4.0 # via requests -chz==0.3.0 - # via gpt-oss click==8.1.7 # via # black @@ -105,7 +99,6 @@ click==8.1.7 # ray # schemathesis # typer - # uvicorn colorama==0.4.6 # via # perceptron @@ -117,11 +110,11 @@ colorlog==6.10.1 # via optuna contourpy==1.3.0 # via matplotlib -coverage==7.10.6 +coverage[toml]==7.10.6 # via pytest-cov cramjam==2.9.0 # via fastparquet -cryptography==46.0.5 +cryptography==46.0.7 # via # azure-identity # azure-storage-blob @@ -131,28 +124,28 @@ cuda-bindings==13.0.3 # via torch cuda-pathfinder==1.3.3 # via cuda-bindings -cuda-toolkit==13.0.2 +cuda-toolkit[cublas,cudart,cufft,cufile,cupti,curand,cusolver,cusparse,nvjitlink,nvrtc,nvtx]==13.0.2 # via torch cupy-cuda12x==13.6.0 # via ray cycler==0.12.1 # via matplotlib datamodel-code-generator==0.26.3 - # via -r requirements/test/cuda.in + # via -r test/cuda.in dataproperty==1.0.1 # via # pytablewriter # tabledata datasets==3.3.0 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # evaluate # lm-eval # mteb decorator==5.1.1 # via librosa -decord==0.6.0 - # via -r requirements/test/cuda.in +decord==0.6.0 ; platform_machine == "x86_64" + # via -r test/cuda.in dill==0.3.8 # via # datasets @@ -163,13 +156,11 @@ distlib==0.3.9 # via virtualenv dnspython==2.7.0 # via email-validator -docker==7.1.0 - # via gpt-oss docopt==0.6.2 # via num2words einops==0.8.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # encodec # vector-quantize-pytorch # vocos @@ -183,21 +174,14 @@ et-xmlfile==2.0.0 # via openpyxl evaluate==0.4.3 # via lm-eval -fastapi==0.128.0 - # via - # -c requirements/common.txt - # gpt-oss fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 # via cupy-cuda12x -fastsafetensors==0.2.2 - # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in +fastsafetensors==0.2.2 ; platform_machine == "x86_64" + # via -r test/cuda.in filelock==3.16.1 # via - # -c requirements/common.txt # blobfile # datasets # huggingface-hub @@ -214,7 +198,7 @@ frozenlist==1.5.0 # via # aiohttp # aiosignal -fsspec==2024.12.0 +fsspec[http]==2024.12.0 # via # datasets # evaluate @@ -224,7 +208,7 @@ fsspec==2024.12.0 ftfy==6.3.1 # via open-clip-torch genai-perf==0.0.16 - # via -r requirements/test/cuda.in + # via -r test/cuda.in genson==1.3.0 # via datamodel-code-generator google-api-core==2.24.2 @@ -250,23 +234,19 @@ google-resumable-media==2.7.2 # via google-cloud-storage googleapis-common-protos==1.70.0 # via google-api-core -gpt-oss==0.0.8 - # via -r requirements/test/cuda.in graphql-core==3.2.6 # via hypothesis-graphql greenlet==3.2.3 # via sqlalchemy grpcio==1.78.0 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # grpcio-reflection # ray grpcio-reflection==1.78.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in h11==0.14.0 - # via - # httpcore - # uvicorn + # via httpcore h2==4.3.0 # via httpx harfile==0.3.0 @@ -277,13 +257,11 @@ hiredis==3.0.0 # via tensorizer hpack==4.1.0 # via h2 -html2text==2025.4.15 - # via gpt-oss httpcore==1.0.6 # via httpx -httpx==0.27.2 +httpx[http2]==0.27.2 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # huggingface-hub # perceptron # schemathesis @@ -322,7 +300,7 @@ idna==3.10 # requests # yarl imagehash==4.3.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in imageio==2.37.0 # via scikit-image importlib-metadata==8.7.0 @@ -331,8 +309,8 @@ inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 # via pytest -instanttensor==0.1.5 - # via -r requirements/test/cuda.in +instanttensor==0.1.5 ; platform_machine == "x86_64" + # via -r test/cuda.in isodate==0.7.2 # via azure-storage-blob isoduration==20.11.0 @@ -346,7 +324,7 @@ jinja2==3.1.6 # lm-eval # torch jiwer==3.0.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in jmespath==1.0.1 # via # boto3 @@ -358,7 +336,7 @@ joblib==1.4.2 # scikit-learn jsonpointer==3.0.0 # via jsonschema -jsonschema==4.23.0 +jsonschema[format]==4.23.0 # via # hypothesis-jsonschema # mistral-common @@ -369,7 +347,7 @@ jsonschema-specifications==2024.10.1 junit-xml==1.9 # via schemathesis kaldi-native-fbank==1.22.3 - # via -r requirements/test/cuda.in + # via -r test/cuda.in kaleido==0.2.1 # via genai-perf kiwisolver==1.4.7 @@ -381,15 +359,14 @@ lazy-loader==0.4 libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in llvmlite==0.47.0 # via numba -lm-eval==0.4.12 - # via -r requirements/test/cuda.in +lm-eval[api]==0.4.12 + # via -r test/cuda.in lxml==5.3.0 # via # blobfile - # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -401,7 +378,7 @@ markupsafe==3.0.1 # mako # werkzeug matplotlib==3.9.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in mbstrdecoder==1.1.3 # via # dataproperty @@ -409,10 +386,10 @@ mbstrdecoder==1.1.3 # typepy mdurl==0.1.2 # via markdown-it-py -mistral-common==1.11.2 +mistral-common[audio,image,opencv,soundfile,soxr]==1.11.2 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in + # mistral-common more-itertools==10.5.0 # via lm-eval mpmath==1.3.0 @@ -427,8 +404,8 @@ msgpack==1.1.0 # via # librosa # ray -mteb==2.8.3 - # via -r requirements/test/cuda.in +mteb[bm25s]==2.8.3 + # via -r test/cuda.in multidict==6.1.0 # via # aiohttp @@ -446,15 +423,14 @@ networkx==3.2.1 nltk==3.9.1 # via rouge-score num2words==0.5.14 - # via -r requirements/test/cuda.in + # via -r test/cuda.in numba==0.65.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # librosa numpy==2.2.6 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # accelerate # albumentations # bitsandbytes @@ -538,26 +514,20 @@ nvidia-nvshmem-cu13==3.4.5 nvidia-nvtx==13.0.85 # via cuda-toolkit open-clip-torch==2.32.0 - # via -r requirements/test/cuda.in -openai-harmony==0.0.4 - # via - # -c requirements/common.txt - # gpt-oss + # via -r test/cuda.in opencensus==0.11.4 # via ray opencensus-context==0.1.3 # via opencensus opencv-python-headless==4.13.0.90 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # albumentations # mistral-common openpyxl==3.1.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in opentelemetry-api==1.35.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions @@ -567,7 +537,6 @@ opentelemetry-proto==1.35.0 # via ray opentelemetry-sdk==1.35.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # ray opentelemetry-semantic-conventions==0.56b0 @@ -613,9 +582,9 @@ pathvalidate==3.2.1 patsy==1.0.1 # via statsmodels peft==0.18.1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in perceptron==0.1.4 - # via -r requirements/test/cuda.in + # via -r test/cuda.in perf-analyzer==0.1.0 # via genai-perf pillow==10.4.0 @@ -636,7 +605,7 @@ platformdirs==4.3.6 # virtualenv plotly==5.24.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf pluggy==1.5.0 # via @@ -649,10 +618,9 @@ pooch==1.8.2 portalocker==2.10.1 # via sacrebleu pqdm==0.2.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in prometheus-client==0.22.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # ray propcache==0.2.0 @@ -663,7 +631,6 @@ proto-plus==1.26.1 # via google-api-core protobuf==6.33.6 # via - # -c requirements/common.txt # google-api-core # googleapis-common-protos # grpcio-reflection @@ -696,27 +663,25 @@ pycparser==2.22 # via cffi pycryptodomex==3.22.0 # via blobfile -pydantic==2.12.0 +pydantic[email]==2.12.0 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # albumentations # datamodel-code-generator - # fastapi - # gpt-oss # mistral-common # mteb - # openai-harmony # pydantic-extra-types # ray pydantic-core==2.41.1 # via pydantic -pydantic-extra-types==2.10.5 +pydantic-extra-types[pycountry]==2.10.5 # via mistral-common pygments==2.18.0 # via rich -pyjwt==2.11.0 - # via msal +pyjwt[crypto]==2.11.0 + # via + # msal + # pyjwt pyparsing==3.2.0 # via matplotlib pyrate-limiter==3.7.0 @@ -727,7 +692,7 @@ pytablewriter==1.2.0 # via lm-eval pytest==8.3.5 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # buildkite-test-collector # genai-perf # pytest-asyncio @@ -740,21 +705,21 @@ pytest==8.3.5 # pytest-timeout # schemathesis pytest-asyncio==0.24.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-cov==6.3.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-forked==1.6.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-mock==3.14.0 # via genai-perf pytest-rerunfailures==14.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-shard==0.1.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-subtests==0.14.1 # via schemathesis pytest-timeout==2.3.1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in python-dateutil==2.9.0.post0 # via # arrow @@ -790,8 +755,8 @@ pyyaml==6.0.2 # vocos rapidfuzz==3.12.1 # via jiwer -ray==2.48.0 - # via -r requirements/test/cuda.in +ray[cgraph,default]==2.48.0 + # via -r test/cuda.in redis==5.2.0 # via tensorizer referencing==0.35.1 @@ -807,15 +772,12 @@ regex==2026.2.28 # transformers requests==2.32.3 # via - # -c requirements/common.txt # azure-core # buildkite-test-collector # datasets - # docker # evaluate # google-api-core # google-cloud-storage - # gpt-oss # lm-eval # mistral-common # msal @@ -846,8 +808,8 @@ rpds-py==0.20.1 # referencing rsa==4.9.1 # via google-auth -runai-model-streamer==0.15.7 - # via -r requirements/test/cuda.in +runai-model-streamer[azure,gcs,s3]==0.15.7 + # via -r test/cuda.in runai-model-streamer-azure==0.15.7 # via runai-model-streamer runai-model-streamer-gcs==0.15.7 @@ -860,7 +822,6 @@ sacrebleu==2.4.3 # via lm-eval safetensors==0.7.0 # via - # -c requirements/common.txt # accelerate # open-clip-torch # peft @@ -868,7 +829,7 @@ safetensors==0.7.0 # timm # transformers schemathesis==3.39.15 - # via -r requirements/test/cuda.in + # via -r test/cuda.in scikit-image==0.25.2 # via albumentations scikit-learn==1.5.2 @@ -891,23 +852,17 @@ scipy==1.13.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in sentence-transformers==5.2.0 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # mteb -setuptools==77.0.3 - # via - # -c requirements/common.txt - # pytablewriter - # torch shellingham==1.5.4 # via # perceptron # typer six==1.16.0 # via - # -c requirements/common.txt # junit-xml # opencensus # python-dateutil @@ -923,7 +878,7 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.12.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf # librosa # mistral-common @@ -939,15 +894,12 @@ sqlitedict==2.1.0 # via lm-eval starlette==0.50.0 # via - # fastapi # schemathesis # starlette-testclient starlette-testclient==0.4.1 # via schemathesis statsmodels==0.14.4 # via genai-perf -structlog==25.4.0 - # via gpt-oss sympy==1.13.3 # via # einx @@ -957,46 +909,39 @@ tabledata==1.3.3 tabulate==0.9.0 # via sacrebleu tblib==3.1.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in tcolorpy==0.1.6 # via pytablewriter tenacity==9.1.2 # via - # gpt-oss # lm-eval # plotly tensorizer==2.10.1 - # via -r requirements/test/cuda.in -termcolor==3.1.0 - # via gpt-oss + # via -r test/cuda.in threadpoolctl==3.5.0 # via scikit-learn tifffile==2025.3.30 # via scikit-image tiktoken==0.12.0 # via - # -c requirements/common.txt - # gpt-oss # lm-eval # mistral-common timm==1.0.17 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # open-clip-torch # segmentation-models-pytorch tokenizers==0.22.2 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # transformers tomli==2.2.1 # via schemathesis tomli-w==1.2.0 # via schemathesis -torch==2.11.0+cu130 +torch==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # accelerate # bitsandbytes # encodec @@ -1012,16 +957,14 @@ torch==2.11.0+cu130 # torchvision # vector-quantize-pytorch # vocos -torchaudio==2.11.0+cu130 +torchaudio==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # encodec # vocos -torchvision==0.26.0+cu130 +torchvision==0.26.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # open-clip-torch # segmentation-models-pytorch # timm @@ -1042,19 +985,18 @@ tqdm==4.67.3 # transformers transformers==5.5.3 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf # peft # sentence-transformers # transformers-stream-generator transformers-stream-generator==0.0.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in triton==3.6.0 # via torch tritonclient==2.64.0 - # via -r requirements/test/cuda.in -typepy==1.3.2 + # via -r test/cuda.in +typepy[datetime]==1.3.2 # via # dataproperty # pytablewriter @@ -1069,15 +1011,12 @@ types-python-dateutil==2.9.0.20241206 # via arrow typing-extensions==4.15.0 # via - # -c requirements/common.txt # aiosignal # albumentations # alembic # azure-core # azure-identity # azure-storage-blob - # chz - # fastapi # grpcio # huggingface-hub # librosa @@ -1107,18 +1046,15 @@ urllib3==2.2.3 # via # blobfile # botocore - # docker # requests # responses # tritonclient -uvicorn==0.35.0 - # via gpt-oss vector-quantize-pytorch==1.21.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in virtualenv==20.31.2 # via ray vocos==0.1.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in wcwidth==0.2.13 # via ftfy webcolors==24.11.1