Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion requirements/common.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
1 change: 1 addition & 0 deletions requirements/test.in
Original file line number Diff line number Diff line change
Expand Up @@ -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'
47 changes: 38 additions & 9 deletions requirements/test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -907,6 +925,7 @@ requests==2.32.3
# evaluate
# google-api-core
# google-cloud-storage
# gpt-oss
# huggingface-hub
# lightly
# lm-eval
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand All @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down