diff --git a/requirements/common.txt b/requirements/common.txt index 1530e5a09e75..a52745f69870 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -49,4 +49,3 @@ pybase64 # fast base64 implementation cbor2 # Required for cross-language serialization of hashable objects setproctitle # Used to set process names for better debugging and monitoring openai-harmony >= 0.0.3 # Required for gpt-oss -gpt-oss >= 0.0.7 diff --git a/requirements/test.in b/requirements/test.in index ef21d6db5b4f..66fc2eeae714 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -56,3 +56,4 @@ fastsafetensors>=0.1.10 pydantic>=2.10 # 2.9 leads to error on python 3.10 decord==0.6.0 terratorch @ git+https://github.com/IBM/terratorch.git@1.1.rc3 # required for PrithviMAE test +gpt-oss >= 0.0.7; python_version > '3.11' \ No newline at end of file diff --git a/requirements/test.txt b/requirements/test.txt index 9cab85ce0ef6..795fa1b7ad08 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -10,18 +10,19 @@ aenum==3.1.16 # via lightly affine==2.4.0 # via rasterio -aiohappyeyeballs==2.4.3 +aiohappyeyeballs==2.6.1 # via aiohttp -aiohttp==3.10.11 +aiohttp==3.13.0 # via # aiohttp-cors # datasets # fsspec + # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 # via ray -aiosignal==1.3.1 +aiosignal==1.4.0 # via aiohttp albucore==0.0.16 # via terratorch @@ -103,6 +104,8 @@ 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 @@ -173,7 +176,9 @@ distlib==0.3.9 dnspython==2.7.0 # via email-validator docker==7.1.0 - # via mlflow + # via + # gpt-oss + # mlflow docopt==0.6.2 # via num2words docstring-parser==0.17.0 @@ -199,7 +204,9 @@ eval-type-backport==0.2.2 evaluate==0.4.3 # via lm-eval fastapi==0.116.1 - # via mlflow-skinny + # via + # gpt-oss + # mlflow-skinny fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 @@ -274,6 +281,8 @@ 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.in graphene==3.4.3 # via mlflow graphql-core==3.2.6 @@ -301,6 +310,8 @@ hf-xet==1.1.7 # via huggingface-hub hiredis==3.0.0 # via tensorizer +html2text==2025.4.15 + # via gpt-oss httpcore==1.0.6 # via httpx httpx==0.27.2 @@ -435,6 +446,7 @@ lm-eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b772215 lxml==5.3.0 # via # blobfile + # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -604,6 +616,8 @@ omegaconf==2.3.0 # lightning open-clip-torch==2.32.0 # via -r requirements/test.in +openai-harmony==0.0.4 + # via gpt-oss opencensus==0.11.4 # via ray opencensus-context==0.1.3 @@ -724,7 +738,9 @@ prometheus-client==0.22.0 # opentelemetry-exporter-prometheus # ray propcache==0.2.0 - # via yarl + # via + # aiohttp + # yarl proto-plus==1.26.1 # via google-api-core protobuf==5.28.3 @@ -773,10 +789,12 @@ pydantic==2.11.7 # albumentations # datamodel-code-generator # fastapi + # gpt-oss # lightly # mistral-common # mlflow-skinny # mteb + # openai-harmony # pydantic-extra-types # ray pydantic-core==2.33.2 @@ -907,6 +925,7 @@ requests==2.32.3 # evaluate # google-api-core # google-cloud-storage + # gpt-oss # huggingface-hub # lightly # lm-eval @@ -1052,6 +1071,8 @@ 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 @@ -1064,14 +1085,17 @@ tblib==3.1.0 # via -r requirements/test.in tcolorpy==0.1.6 # via pytablewriter -tenacity==9.0.0 +tenacity==9.1.2 # via + # gpt-oss # lm-eval # plotly tensorboardx==2.6.4 # via lightning tensorizer==2.10.1 # via -r requirements/test.in +termcolor==3.1.0 + # via gpt-oss terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e # via -r requirements/test.in threadpoolctl==3.5.0 @@ -1080,8 +1104,9 @@ tifffile==2025.3.30 # via # scikit-image # terratorch -tiktoken==0.7.0 +tiktoken==0.12.0 # via + # gpt-oss # lm-eval # mistral-common timm==1.0.17 @@ -1201,8 +1226,10 @@ typeshed-client==2.8.2 # via jsonargparse typing-extensions==4.12.2 # via + # aiosignal # albumentations # alembic + # chz # fastapi # graphene # huggingface-hub @@ -1242,7 +1269,9 @@ urllib3==2.2.3 # responses # tritonclient uvicorn==0.35.0 - # via mlflow-skinny + # via + # gpt-oss + # mlflow-skinny vector-quantize-pytorch==1.21.2 # via -r requirements/test.in virtualenv==20.31.2