diff --git a/requirements/rocm-test.in b/requirements/rocm-test.in index 856fab7e9..b8978f1f2 100644 --- a/requirements/rocm-test.in +++ b/requirements/rocm-test.in @@ -1,3 +1,5 @@ +-r common.txt + # testing pytest tensorizer==2.10.1 diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index dd4c7c24f..d5afde3c8 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -15,6 +15,7 @@ aiohappyeyeballs==2.6.1 aiohttp==3.13.3 # via # -c requirements/common.txt + # -r requirements/common.txt # aiohttp-cors # fsspec # gpt-oss @@ -38,20 +39,31 @@ annotated-doc==0.0.4 # typer annotated-types==0.7.0 # via pydantic +anthropic==0.89.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt antlr4-python3-runtime==4.9.3 # via # hydra-core # omegaconf -anyio==4.6.2.post1 +anyio==4.13.0 # via + # anthropic # httpx + # mcp + # openai + # sse-starlette # starlette + # watchfiles arctic-inference==0.1.1 # via -r requirements/rocm-test.in argcomplete==3.6.3 # via datamodel-code-generator arrow==1.4.0 # via isoduration +astor==0.8.1 + # via depyf attrs==26.1.0 # via # aiohttp @@ -83,6 +95,8 @@ bitsandbytes==0.49.2 # lightning black==26.3.1 # via datamodel-code-generator +blake3==1.0.8 + # via -r requirements/common.txt blobfile==3.0.0 # via -r requirements/rocm-test.in bm25s==0.2.13 @@ -99,6 +113,10 @@ bounded-pool-executor==0.0.3 # via pqdm buildkite-test-collector==0.1.9 # via -r requirements/rocm-test.in +cachetools==7.0.5 + # via -r requirements/common.txt +cbor2==5.9.0 + # via -r requirements/common.txt certifi==2026.2.25 # via # fiona @@ -132,6 +150,7 @@ click==8.3.1 # nltk # rasterio # ray + # rich-toolkit # schemathesis # typer # uvicorn @@ -142,6 +161,8 @@ cligj==0.7.2 # via # fiona # rasterio +cloudpickle==3.1.2 + # via -r requirements/common.txt colorama==0.4.6 # via # perceptron @@ -151,6 +172,10 @@ colorful==0.5.8 # via ray colorlog==6.10.1 # via optuna +compressed-tensors==0.14.0.1 + # via + # -c requirements/common.txt + # -r requirements/common.txt contourpy==1.3.3 # via matplotlib coverage==7.13.5 @@ -182,24 +207,42 @@ decorator==5.2.1 # via librosa decord==0.6.0 # via -r requirements/rocm-test.in +depyf==0.20.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt diffusers==0.37.0 # via terratorch dill==0.3.8 # via # datasets + # depyf # evaluate # lm-eval # multiprocess +diskcache==5.6.3 + # via + # -c requirements/common.txt + # -r requirements/common.txt distlib==0.4.0 # via virtualenv +distro==1.9.0 + # via + # anthropic + # openai +dnspython==2.8.0 + # via email-validator docker==7.1.0 # via gpt-oss docopt==0.6.2 # via num2words docstring-parser==0.17.0 - # via jsonargparse + # via + # anthropic + # jsonargparse einops==0.8.2 # via + # -r requirements/common.txt # -r requirements/rocm-test.in # encodec # terratorch @@ -208,6 +251,10 @@ einops==0.8.2 # vocos einx==0.4.2 # via vector-quantize-pytorch +email-validator==2.3.0 + # via + # fastapi + # pydantic encodec==0.1.1 # via vocos et-xmlfile==2.0.0 @@ -217,7 +264,15 @@ evaluate==0.4.6 fastapi==0.135.2 # via # -c requirements/common.txt + # -r requirements/common.txt # gpt-oss + # model-hosting-container-standards +fastapi-cli==0.0.24 + # via fastapi +fastapi-cloud-cli==0.15.1 + # via fastapi-cli +fastar==0.9.0 + # via fastapi-cloud-cli fastparquet==2026.3.0 # via genai-perf fastsafetensors==0.2.2 @@ -225,6 +280,7 @@ fastsafetensors==0.2.2 filelock==3.25.2 # via # -c requirements/common.txt + # -r requirements/common.txt # blobfile # datasets # diffusers @@ -264,6 +320,10 @@ genson==1.3.0 # via datamodel-code-generator geopandas==1.1.3 # via terratorch +gguf==0.18.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt gitdb==4.0.12 # via gitpython gitpython==3.1.46 @@ -290,7 +350,10 @@ google-crc32c==1.8.0 google-resumable-media==2.8.0 # via google-cloud-storage googleapis-common-protos==1.73.0 - # via google-api-core + # via + # google-api-core + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http gpt-oss==0.0.8 # via -r requirements/rocm-test.in graphql-core==3.2.8 @@ -302,6 +365,7 @@ grpcio==1.78.0 # -c requirements/rocm.txt # -r requirements/rocm-test.in # grpcio-reflection + # opentelemetry-exporter-otlp-proto-grpc # ray # tensorboard grpcio-reflection==1.78.0 @@ -328,12 +392,22 @@ html2text==2025.4.15 # via gpt-oss httpcore==1.0.9 # via httpx +httptools==0.7.1 + # via uvicorn httpx==0.27.2 # via # -r requirements/rocm-test.in + # anthropic # diffusers + # fastapi + # fastapi-cloud-cli + # mcp + # model-hosting-container-standards + # openai # perceptron # schemathesis +httpx-sse==0.4.3 + # via mcp huggingface-hub==0.36.2 # via # -r requirements/rocm-test.in @@ -370,10 +444,13 @@ hypothesis-jsonschema==0.23.1 idna==3.11 # via # anyio + # email-validator # httpx # jsonschema # requests # yarl +ijson==3.5.0 + # via -r requirements/common.txt imagehash==4.3.2 # via -r requirements/rocm-test.in imageio==2.37.3 @@ -390,6 +467,8 @@ iniconfig==2.3.0 # via pytest instanttensor==0.1.6 # via -r requirements/rocm-test.in +interegular==0.3.3 + # via lm-format-enforcer isodate==0.7.2 # via azure-storage-blob isoduration==20.11.0 @@ -399,15 +478,21 @@ isort==8.0.1 jinja2==3.1.6 # via # datamodel-code-generator + # fastapi # genai-perf # lm-eval # torch +jiter==0.13.0 + # via + # anthropic + # openai jiwer==4.0.0 # via -r requirements/rocm-test.in jmespath==1.1.0 # via # boto3 # botocore + # model-hosting-container-standards joblib==1.5.3 # via # librosa @@ -426,6 +511,7 @@ jsonpointer==3.1.0 jsonschema==4.26.0 # via # hypothesis-jsonschema + # mcp # mistral-common # ray # schemathesis @@ -443,6 +529,10 @@ kornia==0.8.2 # via torchgeo kornia-rs==0.1.10 # via kornia +lark==1.2.2 + # via + # -c requirements/common.txt + # -r requirements/common.txt lazy-loader==0.4 # via # librosa @@ -466,14 +556,24 @@ lightning-utilities==0.15.3 # lightning # pytorch-lightning # torchmetrics +llguidance==1.3.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt llvmlite==0.44.0 # via numba lm-eval==0.4.11 # via -r requirements/rocm-test.in +lm-format-enforcer==0.11.3 + # via + # -c requirements/common.txt + # -r requirements/common.txt logistro==2.0.1 # via # choreographer # kaleido +loguru==0.7.3 + # via compressed-tensors lxml==6.0.2 # via # blobfile @@ -500,12 +600,19 @@ mbstrdecoder==1.1.4 # dataproperty # pytablewriter # typepy +mcp==1.27.0 + # via -r requirements/common.txt mdurl==0.1.2 # via markdown-it-py mistral-common==1.10.0 # via # -c requirements/common.txt + # -r requirements/common.txt # -r requirements/rocm-test.in +model-hosting-container-standards==0.1.14 + # via + # -c requirements/common.txt + # -r requirements/common.txt more-itertools==10.8.0 # via # inflect @@ -522,6 +629,8 @@ msgpack==1.1.2 # via # librosa # ray +msgspec==0.20.0 + # via -r requirements/common.txt mteb==2.11.5 # via -r requirements/rocm-test.in multidict==6.7.1 @@ -541,6 +650,8 @@ networkx==3.6.1 # via # scikit-image # torch +ninja==1.13.0 + # via -r requirements/common.txt nltk==3.9.3 # via rouge-score num2words==0.5.14 @@ -555,6 +666,7 @@ numkong==7.1.1 # via albucore numpy==2.2.6 # via + # -r requirements/common.txt # -r requirements/rocm-test.in # accelerate # albucore @@ -572,6 +684,7 @@ numpy==2.2.6 # fastparquet # genai-perf # geopandas + # gguf # h5py # imagehash # imageio @@ -620,15 +733,21 @@ numpy==2.2.6 # tritonclient # vocos # xarray + # xgrammar omegaconf==2.3.0 # via # hydra-core # lightning open-clip-torch==2.32.0 # via -r requirements/rocm-test.in +openai==2.30.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt openai-harmony==0.0.8 # via # -c requirements/common.txt + # -r requirements/common.txt # gpt-oss opencensus==0.11.4 # via ray @@ -637,6 +756,7 @@ opencensus-context==0.1.3 opencv-python-headless==4.13.0.92 # via # -c requirements/common.txt + # -r requirements/common.txt # -r requirements/rocm-test.in # albumentations # mistral-common @@ -645,26 +765,59 @@ openpyxl==3.1.5 opentelemetry-api==1.40.0 # via # -c requirements/common.txt + # -r requirements/common.txt + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions +opentelemetry-exporter-otlp==1.40.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt +opentelemetry-exporter-otlp-proto-common==1.40.0 + # via + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http +opentelemetry-exporter-otlp-proto-grpc==1.40.0 + # via opentelemetry-exporter-otlp +opentelemetry-exporter-otlp-proto-http==1.40.0 + # via opentelemetry-exporter-otlp opentelemetry-exporter-prometheus==0.61b0 # via ray opentelemetry-proto==1.40.0 - # via ray + # via + # opentelemetry-exporter-otlp-proto-common + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http + # ray opentelemetry-sdk==1.40.0 # via # -c requirements/common.txt + # -r requirements/common.txt + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus + # opentelemetry-semantic-conventions-ai # ray opentelemetry-semantic-conventions==0.61b0 - # via opentelemetry-sdk + # via + # opentelemetry-sdk + # opentelemetry-semantic-conventions-ai +opentelemetry-semantic-conventions-ai==0.5.1 + # via + # -c requirements/common.txt + # -r requirements/common.txt optuna==3.6.1 # via genai-perf orjson==3.11.7 # via # genai-perf # kaleido +outlines-core==0.2.11 + # via + # -c requirements/common.txt + # -r requirements/common.txt packaging==26.0 # via # -c requirements/rocm.txt @@ -682,6 +835,7 @@ packaging==26.0 # lazy-loader # lightning # lightning-utilities + # lm-format-enforcer # matplotlib # optuna # peft @@ -713,6 +867,8 @@ pandas==3.0.1 # tacoreader # torchgeo # xarray +partial-json-parser==0.2.1.1.post7 + # via -r requirements/common.txt pathspec==1.0.4 # via black pathvalidate==3.3.1 @@ -727,6 +883,7 @@ perf-analyzer==0.1.0 # via genai-perf pillow==12.1.1 # via + # -r requirements/common.txt # diffusers # genai-perf # imagehash @@ -768,8 +925,14 @@ pqdm==0.2.0 prometheus-client==0.24.1 # via # -c requirements/common.txt + # -r requirements/common.txt # opentelemetry-exporter-prometheus + # prometheus-fastapi-instrumentator # ray +prometheus-fastapi-instrumentator==7.1.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt propcache==0.4.1 # via # aiohttp @@ -779,6 +942,7 @@ proto-plus==1.27.1 protobuf==6.33.6 # via # -c requirements/common.txt + # -r requirements/common.txt # google-api-core # googleapis-common-protos # grpcio-reflection @@ -791,11 +955,14 @@ protobuf==6.33.6 # wandb psutil==7.2.2 # via + # -r requirements/common.txt # accelerate # peft # tensorizer py==1.11.0 # via pytest-forked +py-cpuinfo==9.0.0 + # via -r requirements/common.txt py-spy==0.4.1 # via ray pyarrow==23.0.1 @@ -808,6 +975,8 @@ pyasn1==0.6.3 # via pyasn1-modules pyasn1-modules==0.4.2 # via google-auth +pybase64==1.4.3 + # via -r requirements/common.txt pycocotools==2.0.11 # via terratorch pycountry==26.2.16 @@ -819,26 +988,44 @@ pycryptodomex==3.23.0 pydantic==2.12.5 # via # -c requirements/common.txt + # -r requirements/common.txt # -r requirements/rocm-test.in # albumentations + # anthropic + # compressed-tensors # datamodel-code-generator # fastapi + # fastapi-cloud-cli # gpt-oss # lightly + # lm-format-enforcer + # mcp # mistral-common + # model-hosting-container-standards # mteb + # openai # openai-harmony # pydantic-extra-types + # pydantic-settings # ray # wandb + # xgrammar pydantic-core==2.41.5 # via pydantic pydantic-extra-types==2.11.1 - # via mistral-common + # via + # fastapi + # mistral-common +pydantic-settings==2.13.1 + # via + # fastapi + # mcp pygments==2.19.2 # via rich pyjwt==2.12.1 - # via msal + # via + # mcp + # msal pyogrio==0.12.1 # via geopandas pyparsing==3.3.2 @@ -898,6 +1085,16 @@ python-dateutil==2.9.0.post0 # typepy python-discovery==1.2.0 # via virtualenv +python-dotenv==1.2.2 + # via + # pydantic-settings + # uvicorn +python-json-logger==4.1.0 + # via -r requirements/common.txt +python-multipart==0.0.22 + # via + # fastapi + # mcp python-rapidjson==1.23 # via tritonclient pytokens==0.4.1 @@ -914,14 +1111,17 @@ pywavelets==1.9.0 # via imagehash pyyaml==6.0.3 # via + # -r requirements/common.txt # accelerate # albumentations # datamodel-code-generator # datasets # genai-perf + # gguf # huggingface-hub # jsonargparse # lightning + # lm-format-enforcer # omegaconf # optuna # peft @@ -931,8 +1131,13 @@ pyyaml==6.0.3 # schemathesis # timm # transformers + # uvicorn # vocos # wandb +pyzmq==27.1.0 + # via + # -c requirements/common.txt + # -r requirements/common.txt rapidfuzz==3.12.1 # via # -r requirements/rocm-test.in @@ -952,6 +1157,7 @@ referencing==0.37.0 # jsonschema-specifications regex==2026.2.28 # via + # -r requirements/common.txt # diffusers # nltk # open-clip-torch @@ -961,12 +1167,14 @@ regex==2026.2.28 requests==2.32.5 # via # -c requirements/common.txt + # -r requirements/common.txt # azure-core # buildkite-test-collector # datasets # diffusers # docker # evaluate + # gguf # google-api-core # google-cloud-storage # gpt-oss @@ -976,6 +1184,7 @@ requests==2.32.5 # mistral-common # msal # mteb + # opentelemetry-exporter-otlp-proto-http # pooch # ray # responses @@ -999,8 +1208,15 @@ rich==14.3.3 # lightning # mteb # perceptron + # rich-toolkit # terratorch # typer +rich-toolkit==0.19.7 + # via + # fastapi-cli + # fastapi-cloud-cli +rignore==0.7.6 + # via fastapi-cloud-cli rioxarray==0.22.0 # via terratorch rouge-score==0.1.2 @@ -1070,12 +1286,20 @@ sentence-transformers==5.3.0 # via # -r requirements/rocm-test.in # mteb +sentencepiece==0.2.1 + # via -r requirements/common.txt sentry-sdk==2.55.0 - # via wandb + # via + # fastapi-cloud-cli + # wandb +setproctitle==1.3.7 + # via -r requirements/common.txt setuptools==79.0.1 # via # -c requirements/common.txt # -c requirements/rocm.txt + # -r requirements/common.txt + # model-hosting-container-standards # pytablewriter # tensorboard # torch @@ -1092,6 +1316,7 @@ simplejson==3.20.2 six==1.17.0 # via # -c requirements/common.txt + # -r requirements/common.txt # junit-xml # lightly # opencensus @@ -1104,8 +1329,9 @@ smmap==5.0.3 # via gitdb sniffio==1.3.1 # via - # anyio + # anthropic # httpx + # openai sortedcontainers==2.4.0 # via hypothesis soundfile==0.13.1 @@ -1124,10 +1350,16 @@ sqlalchemy==2.0.48 # optuna sqlitedict==2.1.0 # via lm-eval +sse-starlette==3.3.4 + # via mcp starlette==0.52.1 # via # fastapi + # mcp + # model-hosting-container-standards + # prometheus-fastapi-instrumentator # schemathesis + # sse-starlette # starlette-testclient starlette-testclient==0.4.1 # via schemathesis @@ -1137,6 +1369,8 @@ stringzilla==4.6.0 # via albucore structlog==25.5.0 # via gpt-oss +supervisor==4.3.0 + # via model-hosting-container-standards sympy==1.14.0 # via # einx @@ -1180,6 +1414,7 @@ tifffile==2026.3.3 tiktoken==0.12.0 # via # -c requirements/common.txt + # -r requirements/common.txt # gpt-oss # lm-eval # mistral-common @@ -1194,6 +1429,7 @@ timm==1.0.17 tokenizers==0.22.0 # via # -c requirements/common.txt + # -r requirements/common.txt # -r requirements/rocm-test.in # transformers tomli==2.4.0 @@ -1212,8 +1448,10 @@ torchmetrics==1.9.0 # torchgeo tqdm==4.67.3 # via + # -r requirements/common.txt # datasets # evaluate + # gguf # huggingface-hub # lightly # lightning @@ -1221,6 +1459,7 @@ tqdm==4.67.3 # mteb # nltk # open-clip-torch + # openai # optuna # peft # pqdm @@ -1233,11 +1472,14 @@ tqdm==4.67.3 transformers==4.57.5 # via # -c requirements/common.txt + # -r requirements/common.txt # -r requirements/rocm-test.in + # compressed-tensors # genai-perf # peft # sentence-transformers # transformers-stream-generator + # xgrammar transformers-stream-generator==0.0.5 # via -r requirements/rocm-test.in tritonclient==2.66.0 @@ -1251,6 +1493,8 @@ typepy==1.3.4 # tabledata typer==0.24.1 # via + # fastapi-cli + # fastapi-cloud-cli # fastsafetensors # perceptron typeshed-client==2.9.0 @@ -1258,9 +1502,12 @@ typeshed-client==2.9.0 typing-extensions==4.15.0 # via # -c requirements/common.txt + # -r requirements/common.txt # aiosignal # albumentations # alembic + # anthropic + # anyio # azure-core # azure-identity # azure-storage-blob @@ -1272,9 +1519,13 @@ typing-extensions==4.15.0 # lightning # lightning-utilities # lm-eval + # mcp # mistral-common # mteb + # openai # opentelemetry-api + # opentelemetry-exporter-otlp-proto-grpc + # opentelemetry-exporter-otlp-proto-http # opentelemetry-sdk # opentelemetry-semantic-conventions # pqdm @@ -1283,6 +1534,7 @@ typing-extensions==4.15.0 # pydantic-extra-types # pytorch-lightning # referencing + # rich-toolkit # sentence-transformers # sqlalchemy # starlette @@ -1292,10 +1544,13 @@ typing-extensions==4.15.0 # typeshed-client # typing-inspection # wandb + # xgrammar typing-inspection==0.4.2 # via # fastapi + # mcp # pydantic + # pydantic-settings tzdata==2025.3 # via arrow uri-template==1.3.0 @@ -1311,7 +1566,14 @@ urllib3==2.6.3 # sentry-sdk # tritonclient uvicorn==0.42.0 - # via gpt-oss + # via + # fastapi + # fastapi-cli + # fastapi-cloud-cli + # gpt-oss + # mcp +uvloop==0.22.1 + # via uvicorn vector-quantize-pytorch==1.28.0 # via -r requirements/rocm-test.in virtualenv==21.2.0 @@ -1320,10 +1582,16 @@ vocos==0.1.0 # via -r requirements/rocm-test.in wandb==0.25.1 # via terratorch +watchfiles==1.1.1 + # via + # -r requirements/common.txt + # uvicorn wcwidth==0.6.0 # via ftfy webcolors==25.10.0 # via jsonschema +websockets==16.0 + # via uvicorn werkzeug==3.1.6 # via # schemathesis @@ -1334,6 +1602,10 @@ wrapt==2.1.2 # via smart-open xarray==2026.2.0 # via rioxarray +xgrammar==0.1.33 + # via + # -c requirements/common.txt + # -r requirements/common.txt xxhash==3.6.0 # via # datasets