# This file was autogenerated by uv via the following command: # uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu129 --python-platform x86_64-manylinux_2_28 --python-version 3.12 absl-py==2.1.0 # via # rouge-score # tensorboard accelerate==1.0.1 # via peft aenum==3.1.16 # via lightly affine==2.4.0 # via rasterio aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 # via # aiohttp-cors # datasets # fsspec # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 # via ray aiosignal==1.4.0 # via aiohttp albucore==0.0.16 # via terratorch albumentations==1.4.6 # via # -r requirements/test.in # terratorch alembic==1.16.4 # via optuna annotated-doc==0.0.4 # via fastapi annotated-types==0.7.0 # via pydantic antlr4-python3-runtime==4.9.3 # via # hydra-core # omegaconf anyio==4.6.2.post1 # via # httpx # starlette arctic-inference==0.1.1 # via -r requirements/test.in argcomplete==3.5.1 # via datamodel-code-generator arrow==1.3.0 # via isoduration attrs==24.2.0 # via # aiohttp # fiona # hypothesis # jsonlines # jsonschema # pytest-subtests # rasterio # referencing audioread==3.0.1 # via librosa backoff==2.2.1 # via # -r requirements/test.in # schemathesis bitsandbytes==0.46.1 # via # -r requirements/test.in # lightning black==24.10.0 # via datamodel-code-generator blobfile==3.0.0 # via -r requirements/test.in bm25s==0.2.13 # via mteb boto3==1.35.57 # via # runai-model-streamer-s3 # tensorizer botocore==1.35.57 # via # boto3 # s3transfer bounded-pool-executor==0.0.3 # via pqdm buildkite-test-collector==0.1.9 # via -r requirements/test.in cachetools==5.5.2 # via google-auth certifi==2024.8.30 # via # fiona # httpcore # httpx # lightly # pyogrio # pyproj # rasterio # requests # sentry-sdk cffi==1.17.1 # via soundfile 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 # click-plugins # cligj # fiona # jiwer # nltk # rasterio # ray # schemathesis # typer # uvicorn # wandb click-plugins==1.1.1.2 # via # fiona # rasterio cligj==0.7.2 # via # fiona # rasterio colorama==0.4.6 # via # perceptron # sacrebleu # schemathesis colorful==0.5.6 # via ray colorlog==6.10.1 # via optuna contourpy==1.3.0 # via matplotlib coverage==7.10.6 # via pytest-cov cramjam==2.9.0 # via fastparquet cuda-bindings==12.9.4 # via torch cuda-pathfinder==1.3.3 # via cuda-bindings cupy-cuda12x==13.6.0 # via ray cycler==0.12.1 # via matplotlib datamodel-code-generator==0.26.3 # via -r requirements/test.in dataproperty==1.0.1 # via # pytablewriter # tabledata datasets==3.3.0 # via # -r requirements/test.in # evaluate # lm-eval # mteb decorator==5.1.1 # via librosa decord==0.6.0 # via -r requirements/test.in diffusers==0.36.0 # via terratorch dill==0.3.8 # via # datasets # evaluate # lm-eval # multiprocess 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 docstring-parser==0.17.0 # via jsonargparse einops==0.8.1 # via # -r requirements/test.in # encodec # terratorch # torchgeo # vector-quantize-pytorch # vocos einx==0.3.0 # via vector-quantize-pytorch email-validator==2.2.0 # via pydantic encodec==0.1.1 # via vocos evaluate==0.4.3 # via lm-eval fastapi==0.128.0 # via gpt-oss fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 # via cupy-cuda12x fastsafetensors==0.2.2 # via -r requirements/test.in filelock==3.16.1 # via # blobfile # datasets # diffusers # huggingface-hub # ray # torch # transformers # virtualenv fiona==1.10.1 # via torchgeo fonttools==4.55.0 # via matplotlib fqdn==1.5.1 # via jsonschema frozendict==2.4.6 # via einx frozenlist==1.5.0 # via # aiohttp # aiosignal fsspec==2024.12.0 # via # datasets # evaluate # fastparquet # huggingface-hub # lightning # pytorch-lightning # tacoreader # torch ftfy==6.3.1 # via open-clip-torch genai-perf==0.0.16 # via -r requirements/test.in genson==1.3.0 # via datamodel-code-generator geopandas==1.0.1 # via terratorch gitdb==4.0.12 # via gitpython gitpython==3.1.44 # via wandb google-api-core==2.24.2 # via # google-cloud-core # google-cloud-storage # opencensus google-auth==2.40.2 # via # google-api-core # google-cloud-core # google-cloud-storage # runai-model-streamer-gcs google-cloud-core==2.4.3 # via google-cloud-storage google-cloud-storage==3.4.0 # via runai-model-streamer-gcs google-crc32c==1.7.1 # via # google-cloud-storage # google-resumable-media 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 graphql-core==3.2.6 # via hypothesis-graphql greenlet==3.2.3 # via sqlalchemy grpcio==1.78.0 # via # grpcio-tools # ray # tensorboard grpcio-tools==1.78.0 # via -r requirements/test.in h11==0.14.0 # via # httpcore # uvicorn h2==4.3.0 # via httpx h5py==3.13.0 # via terratorch harfile==0.3.0 # via schemathesis hf-xet==1.1.7 # via huggingface-hub 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 # via # -r requirements/test.in # diffusers # perceptron # schemathesis huggingface-hub==0.36.2 # via # accelerate # datasets # diffusers # evaluate # open-clip-torch # peft # segmentation-models-pytorch # sentence-transformers # terratorch # timm # tokenizers # transformers # vocos humanize==4.11.0 # via runai-model-streamer hydra-core==1.3.2 # via # lightly # lightning hyperframe==6.1.0 # via h2 hypothesis==6.131.0 # via # hypothesis-graphql # hypothesis-jsonschema # schemathesis hypothesis-graphql==0.11.1 # via schemathesis hypothesis-jsonschema==0.23.1 # via schemathesis idna==3.10 # via # anyio # email-validator # httpx # jsonschema # requests # yarl imagehash==4.3.2 # via -r requirements/test.in imageio==2.37.0 # via scikit-image importlib-metadata==8.7.0 # via # diffusers # opentelemetry-api importlib-resources==6.5.2 # via typeshed-client inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 # via pytest isoduration==20.11.0 # via jsonschema isort==5.13.2 # via datamodel-code-generator jinja2==3.1.6 # via # datamodel-code-generator # genai-perf # lm-eval # torch jiwer==3.0.5 # via -r requirements/test.in jmespath==1.0.1 # via # boto3 # botocore joblib==1.4.2 # via # librosa # nltk # scikit-learn jsonargparse==4.46.0 # via # lightning # terratorch jsonlines==4.0.0 # via lm-eval jsonnet==0.21.0 # via jsonargparse jsonpointer==3.0.0 # via jsonschema jsonschema==4.23.0 # via # hypothesis-jsonschema # mistral-common # ray # schemathesis jsonschema-specifications==2024.10.1 # via jsonschema junit-xml==1.9 # via schemathesis kaleido==0.2.1 # via genai-perf kiwisolver==1.4.7 # via matplotlib kornia==0.8.1 # via torchgeo kornia-rs==0.1.9 # via kornia lazy-loader==0.4 # via # librosa # scikit-image libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 # via -r requirements/test.in lightly==1.5.22 # via # terratorch # torchgeo lightly-utils==0.0.2 # via lightly lightning==2.6.1 # via # terratorch # torchgeo lightning-utilities==0.14.3 # via # lightning # pytorch-lightning # torchmetrics llvmlite==0.44.0 # via numba lm-eval==0.4.11 # via -r requirements/test.in lxml==5.3.0 # via # blobfile # gpt-oss # sacrebleu mako==1.3.10 # via alembic markdown==3.8.2 # via tensorboard markdown-it-py==3.0.0 # via rich markupsafe==3.0.1 # via # jinja2 # mako # werkzeug matplotlib==3.9.2 # via # -r requirements/test.in # lightning # pycocotools # torchgeo mbstrdecoder==1.1.3 # via # dataproperty # pytablewriter # typepy mdurl==0.1.2 # via markdown-it-py mistral-common==1.9.1 # via -r requirements/test.in more-itertools==10.5.0 # via lm-eval mpmath==1.3.0 # via sympy msgpack==1.1.0 # via # librosa # ray mteb==2.1.2 # via -r requirements/test.in multidict==6.1.0 # via # aiohttp # yarl multiprocess==0.70.16 # via # datasets # evaluate mypy-extensions==1.0.0 # via black networkx==3.2.1 # via # scikit-image # torch nltk==3.9.1 # via rouge-score num2words==0.5.14 # via -r requirements/test.in numba==0.61.2 # via # -r requirements/test.in # librosa numpy==2.2.6 # via # -r requirements/test.in # accelerate # albucore # albumentations # bitsandbytes # bm25s # contourpy # cupy-cuda12x # datasets # decord # diffusers # einx # encodec # evaluate # fastparquet # genai-perf # geopandas # h5py # imagehash # imageio # librosa # lightly # lightly-utils # lm-eval # matplotlib # mistral-common # mteb # numba # opencv-python-headless # optuna # pandas # patsy # peft # perceptron # pycocotools # pyogrio # pywavelets # rasterio # rioxarray # rouge-score # runai-model-streamer # sacrebleu # scikit-image # scikit-learn # scipy # segmentation-models-pytorch # shapely # soxr # statsmodels # tensorboard # tensorboardx # tensorizer # terratorch # tifffile # torchgeo # torchmetrics # torchvision # transformers # tritonclient # vocos # xarray nvidia-cublas-cu12==12.9.1.4 # via # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch nvidia-cuda-cupti-cu12==12.9.79 # via torch nvidia-cuda-nvrtc-cu12==12.9.86 # via torch nvidia-cuda-runtime-cu12==12.9.79 # via torch nvidia-cudnn-cu12==9.10.2.21 # via torch nvidia-cufft-cu12==11.4.1.4 # via torch nvidia-cufile-cu12==1.14.1.1 # via torch nvidia-curand-cu12==10.3.10.19 # via torch nvidia-cusolver-cu12==11.7.5.82 # via torch nvidia-cusparse-cu12==12.5.10.65 # via # nvidia-cusolver-cu12 # torch nvidia-cusparselt-cu12==0.7.1 # via torch nvidia-nccl-cu12==2.27.5 # via torch nvidia-nvjitlink-cu12==12.9.86 # via # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 # torch nvidia-nvshmem-cu12==3.4.5 # via torch nvidia-nvtx-cu12==12.9.79 # via torch omegaconf==2.3.0 # via # hydra-core # 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 # via opencensus opencv-python-headless==4.13.0.90 # via # -r requirements/test.in # albucore # albumentations # mistral-common opentelemetry-api==1.35.0 # via # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions opentelemetry-exporter-prometheus==0.56b0 # via ray opentelemetry-proto==1.36.0 # via ray opentelemetry-sdk==1.35.0 # via # opentelemetry-exporter-prometheus # ray opentelemetry-semantic-conventions==0.56b0 # via opentelemetry-sdk optuna==3.6.1 # via genai-perf orjson==3.11.5 # via genai-perf packaging==24.2 # via # accelerate # black # datamodel-code-generator # datasets # evaluate # fastparquet # geopandas # huggingface-hub # hydra-core # kornia # lazy-loader # lightning # lightning-utilities # matplotlib # optuna # peft # plotly # pooch # pyogrio # pytest # pytest-rerunfailures # pytorch-lightning # ray # rioxarray # scikit-image # statsmodels # tensorboard # tensorboardx # torchmetrics # transformers # typepy # wandb # xarray pandas==2.2.3 # via # datasets # evaluate # fastparquet # genai-perf # geopandas # statsmodels # tacoreader # torchgeo # xarray pathspec==0.12.1 # via black pathvalidate==3.2.1 # via pytablewriter patsy==1.0.1 # via statsmodels peft==0.16.0 # via -r requirements/test.in perceptron==0.1.4 # via -r requirements/test.in perf-analyzer==0.1.0 # via genai-perf pillow==10.4.0 # via # diffusers # genai-perf # imagehash # imageio # lightly-utils # matplotlib # mistral-common # perceptron # scikit-image # segmentation-models-pytorch # tensorboard # torchgeo # torchvision platformdirs==4.3.6 # via # black # pooch # virtualenv # wandb plotly==5.24.1 # via genai-perf pluggy==1.5.0 # via # pytest # pytest-cov polars==1.29.0 # via mteb pooch==1.8.2 # via librosa portalocker==2.10.1 # via sacrebleu pqdm==0.2.0 # via -r requirements/test.in prometheus-client==0.22.0 # via # opentelemetry-exporter-prometheus # ray propcache==0.2.0 # via # aiohttp # yarl proto-plus==1.26.1 # via google-api-core protobuf==6.33.2 # via # google-api-core # googleapis-common-protos # grpcio-tools # opentelemetry-proto # proto-plus # ray # tensorboard # tensorboardx # tensorizer # wandb psutil==6.1.0 # via # accelerate # peft # tensorizer py==1.11.0 # via pytest-forked py-spy==0.4.0 # via ray pyarrow==23.0.0 # via # datasets # genai-perf # tacoreader # terratorch pyasn1==0.6.1 # via # pyasn1-modules # rsa pyasn1-modules==0.4.2 # via google-auth pycocotools==2.0.8 # via terratorch pycountry==24.6.1 # via pydantic-extra-types pycparser==2.22 # via cffi pycryptodomex==3.22.0 # via blobfile pydantic==2.12.0 # via # -r requirements/test.in # albumentations # datamodel-code-generator # fastapi # gpt-oss # lightly # mistral-common # mteb # openai-harmony # pydantic-extra-types # ray # wandb pydantic-core==2.41.1 # via pydantic pydantic-extra-types==2.10.5 # via mistral-common pygments==2.18.0 # via rich pyogrio==0.11.0 # via geopandas pyparsing==3.2.0 # via # matplotlib # rasterio pyproj==3.7.1 # via # geopandas # rioxarray # torchgeo pyrate-limiter==3.7.0 # via schemathesis pystemmer==3.0.0 # via mteb pytablewriter==1.2.0 # via lm-eval pytest==8.3.5 # via # -r requirements/test.in # buildkite-test-collector # genai-perf # pytest-asyncio # pytest-cov # pytest-forked # pytest-mock # pytest-rerunfailures # pytest-shard # pytest-subtests # pytest-timeout # schemathesis pytest-asyncio==0.24.0 # via -r requirements/test.in pytest-cov==6.3.0 # via -r requirements/test.in pytest-forked==1.6.0 # via -r requirements/test.in pytest-mock==3.14.0 # via genai-perf pytest-rerunfailures==14.0 # via -r requirements/test.in pytest-shard==0.1.2 # via -r requirements/test.in pytest-subtests==0.14.1 # via schemathesis pytest-timeout==2.3.1 # via -r requirements/test.in python-box==7.3.2 # via terratorch python-dateutil==2.9.0.post0 # via # arrow # botocore # lightly # matplotlib # pandas # typepy python-rapidjson==1.20 # via tritonclient pytorch-lightning==2.5.2 # via # lightly # lightning pytrec-eval-terrier==0.5.7 # via mteb pytz==2024.2 # via # pandas # typepy pywavelets==1.9.0 # via imagehash pyyaml==6.0.2 # via # accelerate # albumentations # datamodel-code-generator # datasets # genai-perf # huggingface-hub # jsonargparse # lightning # omegaconf # optuna # peft # pytorch-lightning # ray # responses # schemathesis # timm # transformers # vocos # wandb rapidfuzz==3.12.1 # via jiwer rasterio==1.4.3 # via # rioxarray # terratorch # torchgeo ray==2.48.0 # via -r requirements/test.in redis==5.2.0 # via tensorizer referencing==0.35.1 # via # jsonschema # jsonschema-specifications regex==2024.9.11 # via # diffusers # nltk # open-clip-torch # sacrebleu # tiktoken # transformers requests==2.32.3 # via # buildkite-test-collector # datasets # diffusers # docker # evaluate # google-api-core # google-cloud-storage # gpt-oss # huggingface-hub # lightly # lm-eval # mistral-common # mteb # pooch # ray # responses # schemathesis # starlette-testclient # tacoreader # tiktoken # transformers # wandb responses==0.25.3 # via genai-perf rfc3339-validator==0.1.4 # via jsonschema rfc3987==1.3.8 # via jsonschema rich==13.9.4 # via # genai-perf # lightning # mteb # perceptron # terratorch # typer rioxarray==0.19.0 # via terratorch rouge-score==0.1.2 # via lm-eval rpds-py==0.20.1 # via # jsonschema # referencing rsa==4.9.1 # via google-auth rtree==1.4.0 # via torchgeo runai-model-streamer==0.15.3 # via -r requirements/test.in runai-model-streamer-gcs==0.15.3 # via runai-model-streamer runai-model-streamer-s3==0.15.3 # via runai-model-streamer s3transfer==0.10.3 # via boto3 sacrebleu==2.4.3 # via lm-eval safetensors==0.4.5 # via # accelerate # diffusers # open-clip-torch # peft # segmentation-models-pytorch # timm # transformers schemathesis==3.39.15 # via -r requirements/test.in scikit-image==0.25.2 # via # albumentations # terratorch scikit-learn==1.5.2 # via # albumentations # librosa # lm-eval # mteb # sentence-transformers # terratorch scipy==1.13.1 # via # albumentations # bm25s # imagehash # librosa # mteb # scikit-image # scikit-learn # sentence-transformers # statsmodels # vocos segmentation-models-pytorch==0.5.0 # via # -r requirements/test.in # terratorch # torchgeo sentence-transformers==5.2.0 # via # -r requirements/test.in # mteb sentry-sdk==2.52.0 # via wandb setuptools==77.0.3 # via # grpcio-tools # lightning-utilities # pytablewriter # tensorboard # torch shapely==2.1.1 # via # geopandas # torchgeo shellingham==1.5.4 # via # perceptron # typer six==1.16.0 # via # junit-xml # lightly # opencensus # python-dateutil # rfc3339-validator # rouge-score smart-open==7.1.0 # via ray smmap==5.0.2 # via gitdb sniffio==1.3.1 # via # anyio # httpx sortedcontainers==2.4.0 # via hypothesis soundfile==0.12.1 # via # -r requirements/test.in # genai-perf # librosa # mistral-common soxr==0.5.0.post1 # via # librosa # mistral-common sqlalchemy==2.0.41 # via # alembic # optuna 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 # torch tabledata==1.3.3 # via pytablewriter tabulate==0.9.0 # via sacrebleu tacoreader==0.5.6 # via terratorch tblib==3.1.0 # via -r requirements/test.in tcolorpy==0.1.6 # via pytablewriter tenacity==9.1.2 # via # gpt-oss # lm-eval # plotly tensorboard==2.20.0 # via terratorch tensorboard-data-server==0.7.2 # via tensorboard tensorboardx==2.6.4 # via lightning tensorizer==2.10.1 # via -r requirements/test.in termcolor==3.1.0 # via # gpt-oss # terratorch terratorch==1.2.2 # via -r requirements/test.in threadpoolctl==3.5.0 # via scikit-learn tifffile==2025.3.30 # via # scikit-image # terratorch tiktoken==0.12.0 # via # gpt-oss # lm-eval # mistral-common timm==1.0.17 # via # -r requirements/test.in # open-clip-torch # segmentation-models-pytorch # terratorch # torchgeo tokenizers==0.22.0 # via # -r requirements/test.in # transformers tomli==2.2.1 # via schemathesis tomli-w==1.2.0 # via schemathesis torch==2.10.0+cu129 # via # -r requirements/test.in # accelerate # bitsandbytes # encodec # kornia # lightly # lightning # mteb # open-clip-torch # peft # pytorch-lightning # runai-model-streamer # segmentation-models-pytorch # sentence-transformers # tensorizer # terratorch # timm # torchaudio # torchgeo # torchmetrics # torchvision # vector-quantize-pytorch # vocos torchaudio==2.10.0+cu129 # via # -r requirements/test.in # encodec # vocos torchgeo==0.7.0 # via terratorch torchmetrics==1.7.4 # via # lightning # pytorch-lightning # terratorch # torchgeo torchvision==0.25.0+cu129 # via # -r requirements/test.in # lightly # open-clip-torch # segmentation-models-pytorch # terratorch # timm # torchgeo tqdm==4.67.3 # via # datasets # evaluate # huggingface-hub # lightly # lightning # lm-eval # mteb # nltk # open-clip-torch # optuna # peft # pqdm # pytorch-lightning # segmentation-models-pytorch # sentence-transformers # tacoreader # terratorch # transformers transformers==4.57.5 # via # -r requirements/test.in # genai-perf # peft # sentence-transformers # transformers-stream-generator transformers-stream-generator==0.0.5 # via -r requirements/test.in triton==3.6.0 # via torch tritonclient==2.64.0 # via -r requirements/test.in typepy==1.3.2 # via # dataproperty # pytablewriter # tabledata typer==0.15.2 # via # fastsafetensors # perceptron types-python-dateutil==2.9.0.20241206 # via arrow typeshed-client==2.8.2 # via jsonargparse typing-extensions==4.15.0 # via # aiosignal # albumentations # alembic # chz # fastapi # grpcio # huggingface-hub # librosa # lightning # lightning-utilities # lm-eval # mistral-common # mteb # opentelemetry-api # opentelemetry-sdk # opentelemetry-semantic-conventions # pqdm # pydantic # pydantic-core # pydantic-extra-types # pytorch-lightning # sentence-transformers # sqlalchemy # starlette # torch # torchgeo # typer # typeshed-client # typing-inspection # wandb typing-inspection==0.4.2 # via pydantic tzdata==2024.2 # via pandas uri-template==1.3.0 # via jsonschema urllib3==2.2.3 # via # blobfile # botocore # docker # lightly # requests # responses # sentry-sdk # tritonclient uvicorn==0.35.0 # via gpt-oss vector-quantize-pytorch==1.21.2 # via -r requirements/test.in virtualenv==20.31.2 # via ray vocos==0.1.0 # via -r requirements/test.in wandb==0.24.2 # via terratorch wcwidth==0.2.13 # via ftfy webcolors==24.11.1 # via jsonschema werkzeug==3.1.3 # via # schemathesis # tensorboard word2number==1.1 # via lm-eval wrapt==1.17.2 # via smart-open xarray==2025.7.1 # via rioxarray xxhash==3.5.0 # via # datasets # evaluate yarl==1.17.1 # via # aiohttp # schemathesis zipp==3.23.0 # via importlib-metadata zstandard==0.23.0 # via lm-eval