From 5c2042b0910b2aa221029140ae66f94a996f8fea Mon Sep 17 00:00:00 2001 From: Stefano Castagnetta Date: Thu, 30 Apr 2026 16:16:15 +0200 Subject: [PATCH 1/2] [CI/Build] Skip terratorch + torchgeo pins while PyPI has `lightning` quarantined MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit PyPI is currently serving the `lightning` project as quarantined; the simple index returns with zero release files exposed. Every published `terratorch` version transitively requires `lightning`, and `torchgeo==0.7.0` requires `lightning[pytorch-extra]`, so `pip-compile` and `pip-compile-rocm` abort with `× No solution found ... Because there are no versions of lightning ...`. Pre-releases do not help (the quarantine hides every version). Comment out the offending pins in `requirements/test/cuda.in` and `requirements/test/rocm.in` and regenerate the lock files. The Prithvi tests get skipped temporarily; revert once PyPI lifts the quarantine. Tracked in #41376. Verified locally that all three pip-compile pre-commit hooks (`pip-compile`, `pip-compile-rocm`, `pip-compile-xpu`) pass on the resulting tree. Signed-off-by: Stefano Castagnetta --- requirements/test/cuda.in | 6 +- requirements/test/cuda.txt | 285 ++----------------------------------- requirements/test/rocm.in | 10 +- requirements/test/rocm.txt | 278 ++---------------------------------- 4 files changed, 37 insertions(+), 542 deletions(-) diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index 73d50104d86a..60be48b00be5 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -61,7 +61,11 @@ fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage instanttensor>=0.1.5 pydantic>=2.12 # 2.11 leads to error on python 3.13 decord==0.6.0; platform_machine == "x86_64" -terratorch >= 1.2.2 # Required for Prithvi tests +# terratorch is temporarily disabled while PyPI has the `lightning` package +# in `quarantined` status (every published terratorch version transitively +# requires `lightning`, so the resolver fails with "no versions of lightning"). +# Re-enable once PyPI lifts the quarantine. Tracked in #41376. +# terratorch >= 1.2.2 # Required for Prithvi tests imagehash # Required for Prithvi tests segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index ffb0d9d00c3d..5c2a26d3dc92 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -1,15 +1,9 @@ # This file was autogenerated by uv via the following command: # uv pip compile requirements/test/cuda.in -c requirements/cuda.txt -o requirements/test/cuda.txt --index-strategy unsafe-best-match --torch-backend cu130 --python-platform x86_64-manylinux_2_28 --python-version 3.12 absl-py==2.1.0 - # via - # rouge-score - # tensorboard + # via rouge-score accelerate==1.13.0 # via peft -aenum==3.1.16 - # via lightly -affine==2.4.0 - # via rasterio aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 @@ -25,22 +19,14 @@ 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/cuda.in - # terratorch + # via -r requirements/test/cuda.in 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 @@ -54,12 +40,10 @@ arrow==1.3.0 attrs==24.2.0 # via # aiohttp - # fiona # hypothesis # jsonlines # jsonschema # pytest-subtests - # rasterio # referencing audioread==3.0.1 # via librosa @@ -78,9 +62,7 @@ backoff==2.2.1 # -r requirements/test/cuda.in # schemathesis bitsandbytes==0.49.2 - # via - # -r requirements/test/cuda.in - # lightning + # via -r requirements/test/cuda.in black==24.10.0 # via datamodel-code-generator blobfile==3.0.0 @@ -103,15 +85,9 @@ cachetools==5.5.2 # via google-auth certifi==2024.8.30 # via - # fiona # httpcore # httpx - # lightly - # pyogrio - # pyproj - # rasterio # requests - # sentry-sdk cffi==2.0.0 # via # cryptography @@ -125,25 +101,12 @@ chz==0.3.0 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 @@ -191,8 +154,6 @@ decorator==5.1.1 # via librosa decord==0.6.0 # via -r requirements/test/cuda.in -diffusers==0.36.0 - # via terratorch dill==0.3.8 # via # datasets @@ -207,14 +168,10 @@ 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/cuda.in # encodec - # terratorch - # torchgeo # vector-quantize-pytorch # vocos einx==0.3.0 @@ -244,13 +201,10 @@ filelock==3.16.1 # -c requirements/common.txt # blobfile # datasets - # diffusers # huggingface-hub # ray # torch # virtualenv -fiona==1.10.1 - # via torchgeo fonttools==4.55.0 # via matplotlib fqdn==1.5.1 @@ -267,9 +221,6 @@ fsspec==2024.12.0 # evaluate # fastparquet # huggingface-hub - # lightning - # pytorch-lightning - # tacoreader # torch ftfy==6.3.1 # via open-clip-torch @@ -277,12 +228,6 @@ genai-perf==0.0.16 # via -r requirements/test/cuda.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 @@ -317,7 +262,6 @@ grpcio==1.78.0 # -r requirements/test/cuda.in # grpcio-reflection # ray - # tensorboard grpcio-reflection==1.78.0 # via -r requirements/test/cuda.in h11==0.14.0 @@ -326,8 +270,6 @@ h11==0.14.0 # uvicorn h2==4.3.0 # via httpx -h5py==3.13.0 - # via terratorch harfile==0.3.0 # via schemathesis hf-xet==1.4.3 @@ -343,7 +285,6 @@ httpcore==1.0.6 httpx==0.27.2 # via # -r requirements/test/cuda.in - # diffusers # huggingface-hub # perceptron # schemathesis @@ -351,23 +292,17 @@ huggingface-hub==1.10.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 @@ -392,11 +327,7 @@ imagehash==4.3.2 imageio==2.37.0 # via scikit-image importlib-metadata==8.7.0 - # via - # diffusers - # opentelemetry-api -importlib-resources==6.5.2 - # via typeshed-client + # via opentelemetry-api inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 @@ -426,14 +357,8 @@ joblib==1.4.2 # 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 @@ -452,10 +377,6 @@ 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 @@ -464,21 +385,6 @@ libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 # via -r requirements/test/cuda.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.47.0 # via numba lm-eval==0.4.11 @@ -490,8 +396,6 @@ lxml==5.3.0 # 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 @@ -500,11 +404,7 @@ markupsafe==3.0.1 # mako # werkzeug matplotlib==3.9.2 - # via - # -r requirements/test/cuda.in - # lightning - # pycocotools - # torchgeo + # via -r requirements/test/cuda.in mbstrdecoder==1.1.3 # via # dataproperty @@ -559,7 +459,6 @@ numpy==2.2.6 # via # -r requirements/test/cuda.in # accelerate - # albucore # albumentations # bitsandbytes # bm25s @@ -567,19 +466,14 @@ numpy==2.2.6 # cupy-cuda12x # datasets # decord - # diffusers # einx # encodec # evaluate # fastparquet # genai-perf - # geopandas - # h5py # imagehash # imageio # librosa - # lightly - # lightly-utils # lm-eval # matplotlib # mistral-common @@ -591,11 +485,7 @@ numpy==2.2.6 # patsy # peft # perceptron - # pycocotools - # pyogrio # pywavelets - # rasterio - # rioxarray # rouge-score # runai-model-streamer # sacrebleu @@ -603,21 +493,14 @@ numpy==2.2.6 # scikit-learn # scipy # segmentation-models-pytorch - # shapely # soxr # statsmodels - # tensorboard - # tensorboardx # tensorizer - # terratorch # tifffile - # torchgeo - # torchmetrics # torchvision # transformers # tritonclient # vocos - # xarray nvidia-cublas==13.1.0.3 # via # cuda-toolkit @@ -657,10 +540,6 @@ nvidia-nvshmem-cu13==3.4.5 # via torch nvidia-nvtx==13.0.85 # via cuda-toolkit -omegaconf==2.3.0 - # via - # hydra-core - # lightning open-clip-torch==2.32.0 # via -r requirements/test/cuda.in openai-harmony==0.0.4 @@ -675,7 +554,6 @@ opencv-python-headless==4.13.0.90 # via # -c requirements/common.txt # -r requirements/test/cuda.in - # albucore # albumentations # mistral-common openpyxl==3.1.5 @@ -710,44 +588,27 @@ packaging==24.2 # 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 @@ -762,25 +623,20 @@ 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 # -r requirements/test/cuda.in @@ -817,10 +673,7 @@ protobuf==6.33.6 # opentelemetry-proto # proto-plus # ray - # tensorboard - # tensorboardx # tensorizer - # wandb psutil==6.1.0 # via # accelerate @@ -834,16 +687,12 @@ 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 @@ -858,13 +707,11 @@ pydantic==2.12.0 # 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 @@ -873,17 +720,8 @@ pygments==2.18.0 # via rich pyjwt==2.11.0 # via msal -pyogrio==0.11.0 - # via geopandas pyparsing==3.2.0 - # via - # matplotlib - # rasterio -pyproj==3.7.1 - # via - # geopandas - # rioxarray - # torchgeo + # via matplotlib pyrate-limiter==3.7.0 # via schemathesis pystemmer==3.0.0 @@ -920,22 +758,15 @@ pytest-subtests==0.14.1 # via schemathesis pytest-timeout==2.3.1 # via -r requirements/test/cuda.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 @@ -952,26 +783,16 @@ pyyaml==6.0.2 # 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/cuda.in redis==5.2.0 @@ -982,7 +803,6 @@ referencing==0.35.1 # jsonschema-specifications regex==2026.2.28 # via - # diffusers # nltk # open-clip-torch # sacrebleu @@ -994,13 +814,11 @@ requests==2.32.3 # azure-core # buildkite-test-collector # datasets - # diffusers # docker # evaluate # google-api-core # google-cloud-storage # gpt-oss - # lightly # lm-eval # mistral-common # msal @@ -1010,9 +828,7 @@ requests==2.32.3 # responses # schemathesis # starlette-testclient - # tacoreader # tiktoken - # wandb responses==0.25.3 # via genai-perf rfc3339-validator==0.1.4 @@ -1022,13 +838,9 @@ rfc3987==1.3.8 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 @@ -1037,8 +849,6 @@ rpds-py==0.20.1 # referencing rsa==4.9.1 # via google-auth -rtree==1.4.0 - # via torchgeo runai-model-streamer==0.15.7 # via -r requirements/test/cuda.in runai-model-streamer-azure==0.15.7 @@ -1054,7 +864,6 @@ sacrebleu==2.4.3 safetensors==0.4.5 # via # accelerate - # diffusers # open-clip-torch # peft # segmentation-models-pytorch @@ -1063,9 +872,7 @@ safetensors==0.4.5 schemathesis==3.39.15 # via -r requirements/test/cuda.in scikit-image==0.25.2 - # via - # albumentations - # terratorch + # via albumentations scikit-learn==1.5.2 # via # albumentations @@ -1073,7 +880,6 @@ scikit-learn==1.5.2 # lm-eval # mteb # sentence-transformers - # terratorch scipy==1.13.1 # via # albumentations @@ -1087,27 +893,16 @@ scipy==1.13.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via - # -r requirements/test/cuda.in - # terratorch - # torchgeo + # via -r requirements/test/cuda.in sentence-transformers==5.2.0 # via # -r requirements/test/cuda.in # mteb -sentry-sdk==2.52.0 - # via wandb setuptools==77.0.3 # via # -c requirements/common.txt - # lightning-utilities # pytablewriter - # tensorboard # torch -shapely==2.1.1 - # via - # geopandas - # torchgeo shellingham==1.5.4 # via # perceptron @@ -1116,15 +911,12 @@ six==1.16.0 # via # -c requirements/common.txt # 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 @@ -1166,8 +958,6 @@ 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/cuda.in tcolorpy==0.1.6 @@ -1177,26 +967,14 @@ tenacity==9.1.2 # 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/cuda.in termcolor==3.1.0 - # via - # gpt-oss - # terratorch -terratorch==1.2.2 - # via -r requirements/test/cuda.in + # via gpt-oss threadpoolctl==3.5.0 # via scikit-learn tifffile==2025.3.30 - # via - # scikit-image - # terratorch + # via scikit-image tiktoken==0.12.0 # via # -c requirements/common.txt @@ -1208,8 +986,6 @@ timm==1.0.17 # -r requirements/test/cuda.in # open-clip-torch # segmentation-models-pytorch - # terratorch - # torchgeo tokenizers==0.22.2 # via # -c requirements/common.txt @@ -1227,21 +1003,14 @@ torch==2.11.0+cu130 # bitsandbytes # encodec # instanttensor - # kornia - # lightly - # lightning # mteb # open-clip-torch # peft - # pytorch-lightning # runai-model-streamer # segmentation-models-pytorch # sentence-transformers # tensorizer - # terratorch # timm - # torchgeo - # torchmetrics # torchvision # vector-quantize-pytorch # vocos @@ -1251,31 +1020,18 @@ torchaudio==2.11.0+cu130 # -r requirements/test/cuda.in # encodec # vocos -torchgeo==0.7.0 - # via terratorch -torchmetrics==1.7.4 - # via - # lightning - # pytorch-lightning - # terratorch - # torchgeo torchvision==0.26.0+cu130 # via # -c requirements/cuda.txt # -r requirements/test/cuda.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 @@ -1283,11 +1039,8 @@ tqdm==4.67.3 # optuna # peft # pqdm - # pytorch-lightning # segmentation-models-pytorch # sentence-transformers - # tacoreader - # terratorch # transformers transformers==5.5.3 # via @@ -1316,8 +1069,6 @@ typer==0.15.2 # transformers types-python-dateutil==2.9.0.20241206 # via arrow -typeshed-client==2.8.2 - # via jsonargparse typing-extensions==4.15.0 # via # -c requirements/common.txt @@ -1332,8 +1083,6 @@ typing-extensions==4.15.0 # grpcio # huggingface-hub # librosa - # lightning - # lightning-utilities # lm-eval # mistral-common # mteb @@ -1344,16 +1093,12 @@ typing-extensions==4.15.0 # 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 @@ -1365,10 +1110,8 @@ urllib3==2.2.3 # blobfile # botocore # docker - # lightly # requests # responses - # sentry-sdk # tritonclient uvicorn==0.35.0 # via gpt-oss @@ -1378,22 +1121,16 @@ virtualenv==20.31.2 # via ray vocos==0.1.0 # via -r requirements/test/cuda.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 + # via schemathesis 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 diff --git a/requirements/test/rocm.in b/requirements/test/rocm.in index b7329b5a9a48..cc04eabf09b2 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -61,7 +61,11 @@ pydantic>=2.12 # 2.11 leads to error on python 3.13 decord==0.6.0 # Prithvi tests -terratorch>=1.2.2 +# terratorch is temporarily disabled while PyPI has the `lightning` package +# in `quarantined` status (every published terratorch version transitively +# requires `lightning`, so the resolver fails with "no versions of lightning"). +# Re-enable once PyPI lifts the quarantine. Tracked in #41376. +# terratorch>=1.2.2 imagehash # Required for Prithvi tests segmentation-models-pytorch>0.4.0 # Required for Prithvi tests @@ -79,5 +83,7 @@ plotly # required for perf comparison html report # ROCm-specific extras (not in CUDA cuda.in) rapidfuzz -torchgeo==0.7.0 +# torchgeo also pulled in `lightning` transitively; disabled for the same +# quarantine reason as terratorch above. Restore once the quarantine clears. +# torchgeo==0.7.0 multiprocess==0.70.16 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index ca33e2d09aa0..801af7db9db1 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -1,15 +1,9 @@ # This file was autogenerated by uv via the following command: # uv pip compile requirements/test/rocm.in -c requirements/rocm.txt -o requirements/test/rocm.txt --index-strategy unsafe-best-match --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13 absl-py==2.4.0 - # via - # rouge-score - # tensorboard + # via rouge-score accelerate==1.13.0 # via peft -aenum==3.1.17 - # via lightly -affine==2.4.0 - # via rasterio aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 @@ -25,12 +19,8 @@ aiohttp-cors==0.8.1 # via ray aiosignal==1.4.0 # via aiohttp -albucore==0.1.2 - # via terratorch albumentations==1.4.6 - # via - # -r requirements/test/rocm.in - # terratorch + # via -r requirements/test/rocm.in alembic==1.18.4 # via optuna annotated-doc==0.0.4 @@ -43,10 +33,6 @@ anthropic==0.93.0 # via # -c requirements/common.txt # -r requirements/test/../common.txt -antlr4-python3-runtime==4.9.3 - # via - # hydra-core - # omegaconf anyio==4.13.0 # via # anthropic @@ -67,11 +53,9 @@ astor==0.8.1 attrs==26.1.0 # via # aiohttp - # fiona # jsonlines # jsonschema # pytest-subtests - # rasterio # referencing audioread==3.0.1 # via librosa @@ -90,9 +74,7 @@ backoff==2.2.1 # -r requirements/test/rocm.in # schemathesis bitsandbytes==0.49.2 - # via - # -r requirements/test/rocm.in - # lightning + # via -r requirements/test/rocm.in black==26.3.1 # via datamodel-code-generator blake3==1.0.8 @@ -119,13 +101,8 @@ cbor2==5.9.0 # via -r requirements/test/../common.txt certifi==2026.2.25 # via - # fiona # httpcore # httpx - # lightly - # pyogrio - # pyproj - # rasterio # requests # sentry-sdk cffi==1.17.1 @@ -143,24 +120,13 @@ chz==0.4.0 click==8.3.1 # via # black - # click-plugins - # cligj - # fiona # jiwer # nltk - # rasterio # ray # rich-toolkit # schemathesis # typer # uvicorn - # wandb -click-plugins==1.1.1.2 - # via fiona -cligj==0.7.2 - # via - # fiona - # rasterio cloudpickle==3.1.2 # via -r requirements/test/../common.txt colorama==0.4.6 @@ -211,8 +177,6 @@ depyf==0.20.0 # via # -c requirements/common.txt # -r requirements/test/../common.txt -diffusers==0.37.0 - # via terratorch dill==0.3.8 # via # datasets @@ -237,16 +201,12 @@ docker==7.1.0 docopt==0.6.2 # via num2words docstring-parser==0.17.0 - # via - # anthropic - # jsonargparse + # via anthropic einops==0.8.2 # via # -r requirements/test/../common.txt # -r requirements/test/rocm.in # encodec - # terratorch - # torchgeo # vector-quantize-pytorch # vocos einx==0.4.2 @@ -283,14 +243,11 @@ filelock==3.25.2 # -r requirements/test/../common.txt # blobfile # datasets - # diffusers # huggingface-hub # python-discovery # ray # torch # virtualenv -fiona==1.10.1 - # via torchgeo fonttools==4.62.1 # via matplotlib fqdn==1.5.1 @@ -307,9 +264,6 @@ fsspec==2025.3.0 # evaluate # fastparquet # huggingface-hub - # lightning - # pytorch-lightning - # tacoreader # torch ftfy==6.3.1 # via open-clip-torch @@ -317,16 +271,10 @@ genai-perf==0.0.16 # via -r requirements/test/rocm.in 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/test/../common.txt -gitdb==4.0.12 - # via gitpython -gitpython==3.1.46 - # via wandb google-api-core==2.30.0 # via # google-cloud-core @@ -366,7 +314,6 @@ grpcio==1.78.0 # grpcio-reflection # opentelemetry-exporter-otlp-proto-grpc # ray - # tensorboard grpcio-reflection==1.78.0 # via # -c requirements/rocm.txt @@ -377,8 +324,6 @@ h11==0.16.0 # uvicorn h2==4.3.0 # via httpx -h5py==3.16.0 - # via terratorch harfile==0.4.0 # via schemathesis hf-xet==1.4.3 @@ -397,7 +342,6 @@ httpx==0.27.2 # via # -r requirements/test/rocm.in # anthropic - # diffusers # fastapi # fastapi-cloud-cli # huggingface-hub @@ -412,23 +356,17 @@ huggingface-hub==1.10.2 # via # accelerate # datasets - # diffusers # evaluate # open-clip-torch # peft # segmentation-models-pytorch # sentence-transformers - # terratorch # timm # tokenizers # transformers # vocos humanize==4.15.0 # via runai-model-streamer -hydra-core==1.3.2 - # via - # lightly - # lightning hyperframe==6.1.0 # via h2 hypothesis==6.151.9 @@ -455,11 +393,7 @@ imagehash==4.3.2 imageio==2.37.3 # via scikit-image importlib-metadata==8.7.1 - # via - # diffusers - # opentelemetry-api -importlib-resources==6.5.2 - # via typeshed-client + # via opentelemetry-api inflect==7.5.0 # via datamodel-code-generator iniconfig==2.3.0 @@ -497,14 +431,8 @@ joblib==1.5.3 # librosa # nltk # scikit-learn -jsonargparse==4.47.0 - # via - # lightning - # terratorch jsonlines==4.0.0 # via lm-eval -jsonnet==0.21.0 - # via jsonargparse jsonpointer==3.1.0 # via jsonschema jsonschema==4.26.0 @@ -524,10 +452,6 @@ kaleido==1.0.0 # via genai-perf kiwisolver==1.5.0 # via matplotlib -kornia==0.8.2 - # via torchgeo -kornia-rs==0.1.10 - # via kornia lark==1.2.2 # via # -c requirements/common.txt @@ -540,21 +464,6 @@ libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 # via -r requirements/test/rocm.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.15.3 - # via - # lightning - # pytorch-lightning - # torchmetrics llguidance==1.3.0 # via # -c requirements/common.txt @@ -580,8 +489,6 @@ lxml==6.0.2 # sacrebleu mako==1.3.10 # via alembic -markdown==3.10.2 - # via tensorboard markdown-it-py==4.0.0 # via rich markupsafe==3.0.3 @@ -590,10 +497,7 @@ markupsafe==3.0.3 # mako # werkzeug matplotlib==3.10.8 - # via - # -r requirements/test/rocm.in - # lightning - # torchgeo + # via -r requirements/test/rocm.in mbstrdecoder==1.1.4 # via # dataproperty @@ -660,14 +564,11 @@ numba==0.65.0 # -c requirements/rocm.txt # -r requirements/test/rocm.in # librosa -numkong==7.1.1 - # via albucore numpy==2.2.6 # via # -r requirements/test/../common.txt # -r requirements/test/rocm.in # accelerate - # albucore # albumentations # bitsandbytes # bm25s @@ -675,20 +576,15 @@ numpy==2.2.6 # cupy-cuda12x # datasets # decord - # diffusers # einx # encodec # evaluate # fastparquet # genai-perf - # geopandas # gguf - # h5py # imagehash # imageio # librosa - # lightly - # lightly-utils # lm-eval # matplotlib # mistral-common @@ -700,12 +596,8 @@ numpy==2.2.6 # patsy # peft # perceptron - # pycocotools - # pyogrio # pytrec-eval-terrier # pywavelets - # rasterio - # rioxarray # rouge-score # runai-model-streamer # sacrebleu @@ -714,27 +606,16 @@ numpy==2.2.6 # scipy # segmentation-models-pytorch # sentence-transformers - # shapely # soundfile # soxr # statsmodels - # tensorboard - # tensorboardx # tensorizer - # terratorch # tifffile - # torchgeo - # torchmetrics # torchvision # transformers # tritonclient # vocos - # xarray # xgrammar -omegaconf==2.3.0 - # via - # hydra-core - # lightning open-clip-torch==2.32.0 # via -r requirements/test/rocm.in openai==2.31.0 @@ -824,46 +705,29 @@ packaging==26.0 # datasets # evaluate # fastparquet - # geopandas # huggingface-hub - # hydra-core # kaleido - # kornia # lazy-loader - # lightning - # lightning-utilities # lm-format-enforcer # matplotlib # optuna # peft # plotly # pooch - # pyogrio # pytest # pytest-rerunfailures - # pytorch-lightning # ray - # rioxarray # scikit-image # statsmodels - # tensorboard - # tensorboardx - # torchmetrics # transformers # typepy - # wandb - # xarray pandas==3.0.1 # via # datasets # evaluate # fastparquet # genai-perf - # geopandas # statsmodels - # tacoreader - # torchgeo - # xarray partial-json-parser==0.2.1.1.post7 # via -r requirements/test/../common.txt pathspec==1.0.4 @@ -881,18 +745,14 @@ perf-analyzer==0.1.0 pillow==12.1.1 # via # -r requirements/test/../common.txt - # diffusers # genai-perf # imagehash # imageio - # lightly-utils # matplotlib # mistral-common # perceptron # scikit-image # segmentation-models-pytorch - # tensorboard - # torchgeo # torchvision platformdirs==4.3.6 # via @@ -900,7 +760,6 @@ platformdirs==4.3.6 # pooch # python-discovery # virtualenv - # wandb plotly==6.6.0 # via # -r requirements/test/rocm.in @@ -946,10 +805,7 @@ protobuf==6.33.6 # opentelemetry-proto # proto-plus # ray - # tensorboard - # tensorboardx # tensorizer - # wandb psutil==7.2.2 # via # -r requirements/test/../common.txt @@ -966,16 +822,12 @@ pyarrow==23.0.1 # via # datasets # genai-perf - # tacoreader - # terratorch pyasn1==0.6.3 # via pyasn1-modules pyasn1-modules==0.4.2 # via google-auth pybase64==1.4.3 # via -r requirements/test/../common.txt -pycocotools==2.0.11 - # via terratorch pycountry==26.2.16 # via pydantic-extra-types pycparser==3.0 @@ -994,7 +846,6 @@ pydantic==2.12.5 # fastapi # fastapi-cloud-cli # gpt-oss - # lightly # lm-format-enforcer # mcp # mistral-common @@ -1005,7 +856,6 @@ pydantic==2.12.5 # pydantic-extra-types # pydantic-settings # ray - # wandb # xgrammar pydantic-core==2.41.5 # via pydantic @@ -1023,17 +873,8 @@ pyjwt==2.12.1 # via # mcp # msal -pyogrio==0.12.1 - # via geopandas pyparsing==3.3.2 - # via - # matplotlib - # rasterio -pyproj==3.7.2 - # via - # geopandas - # rioxarray - # torchgeo + # via matplotlib pyrate-limiter==3.9.0 # via schemathesis pystemmer==3.0.0 @@ -1070,13 +911,10 @@ pytest-subtests==0.14.2 # via schemathesis pytest-timeout==2.3.1 # via -r requirements/test/rocm.in -python-box==7.4.1 - # via terratorch python-dateutil==2.9.0.post0 # via # arrow # botocore - # lightly # matplotlib # pandas # typepy @@ -1096,10 +934,6 @@ python-rapidjson==1.23 # via tritonclient pytokens==0.4.1 # via black -pytorch-lightning==2.6.1 - # via - # lightly - # lightning pytrec-eval-terrier==0.5.10 # via mteb pytz==2026.1.post1 @@ -1116,13 +950,9 @@ pyyaml==6.0.3 # genai-perf # gguf # huggingface-hub - # jsonargparse - # lightning # lm-format-enforcer - # omegaconf # optuna # peft - # pytorch-lightning # ray # responses # schemathesis @@ -1130,7 +960,6 @@ pyyaml==6.0.3 # transformers # uvicorn # vocos - # wandb pyzmq==27.1.0 # via # -c requirements/common.txt @@ -1139,11 +968,6 @@ rapidfuzz==3.12.1 # via # -r requirements/test/rocm.in # jiwer -rasterio==1.5.0 - # via - # rioxarray - # terratorch - # torchgeo ray==2.54.0 # via -r requirements/test/rocm.in redis==7.3.0 @@ -1155,7 +979,6 @@ referencing==0.37.0 regex==2026.2.28 # via # -r requirements/test/../common.txt - # diffusers # nltk # open-clip-torch # sacrebleu @@ -1168,14 +991,12 @@ requests==2.32.5 # azure-core # buildkite-test-collector # datasets - # diffusers # docker # evaluate # gguf # google-api-core # google-cloud-storage # gpt-oss - # lightly # lm-eval # mistral-common # msal @@ -1186,9 +1007,7 @@ requests==2.32.5 # responses # schemathesis # starlette-testclient - # tacoreader # tiktoken - # wandb responses==0.26.0 # via genai-perf rfc3339-validator==0.1.4 @@ -1198,11 +1017,9 @@ rfc3987==1.3.8 rich==14.3.3 # via # genai-perf - # lightning # mteb # perceptron # rich-toolkit - # terratorch # typer rich-toolkit==0.19.7 # via @@ -1210,16 +1027,12 @@ rich-toolkit==0.19.7 # fastapi-cloud-cli rignore==0.7.6 # via fastapi-cloud-cli -rioxarray==0.22.0 - # via terratorch rouge-score==0.1.2 # via lm-eval rpds-py==0.30.0 # via # jsonschema # referencing -rtree==1.4.1 - # via torchgeo runai-model-streamer==0.15.7 # via # -c requirements/rocm.txt @@ -1237,7 +1050,6 @@ sacrebleu==2.6.0 safetensors==0.7.0 # via # accelerate - # diffusers # open-clip-torch # peft # segmentation-models-pytorch @@ -1246,9 +1058,7 @@ safetensors==0.7.0 schemathesis==3.39.15 # via -r requirements/test/rocm.in scikit-image==0.26.0 - # via - # albumentations - # terratorch + # via albumentations scikit-learn==1.8.0 # via # albumentations @@ -1256,7 +1066,6 @@ scikit-learn==1.8.0 # lm-eval # mteb # sentence-transformers - # terratorch scipy==1.17.1 # via # albumentations @@ -1271,10 +1080,7 @@ scipy==1.17.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via - # -r requirements/test/rocm.in - # terratorch - # torchgeo + # via -r requirements/test/rocm.in sentence-transformers==5.3.0 # via # -r requirements/test/rocm.in @@ -1282,9 +1088,7 @@ sentence-transformers==5.3.0 sentencepiece==0.2.1 # via -r requirements/test/../common.txt sentry-sdk==2.55.0 - # via - # fastapi-cloud-cli - # wandb + # via fastapi-cloud-cli setproctitle==1.3.7 # via -r requirements/test/../common.txt setuptools==79.0.1 @@ -1294,12 +1098,7 @@ setuptools==79.0.1 # -r requirements/test/../common.txt # model-hosting-container-standards # pytablewriter - # tensorboard # torch -shapely==2.1.2 - # via - # geopandas - # torchgeo shellingham==1.5.4 # via # perceptron @@ -1311,15 +1110,12 @@ six==1.17.0 # -c requirements/common.txt # -r requirements/test/../common.txt # junit-xml - # lightly # opencensus # python-dateutil # rfc3339-validator # rouge-score smart-open==7.5.1 # via ray -smmap==5.0.3 - # via gitdb sniffio==1.3.1 # via # anthropic @@ -1358,8 +1154,6 @@ starlette-testclient==0.4.1 # via schemathesis statsmodels==0.14.6 # via genai-perf -stringzilla==4.6.0 - # via albucore structlog==25.5.0 # via gpt-oss supervisor==4.3.0 @@ -1372,8 +1166,6 @@ tabledata==1.3.4 # via pytablewriter tabulate==0.10.0 # via sacrebleu -tacoreader==0.5.6 - # via terratorch tblib==3.1.0 # via -r requirements/test/rocm.in tcolorpy==0.1.7 @@ -1382,28 +1174,16 @@ tenacity==9.1.4 # via # gpt-oss # lm-eval -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 # -c requirements/rocm.txt # -r requirements/test/rocm.in termcolor==3.3.0 - # via - # gpt-oss - # terratorch -terratorch==1.2.2 - # via -r requirements/test/rocm.in + # via gpt-oss threadpoolctl==3.6.0 # via scikit-learn tifffile==2026.3.3 - # via - # scikit-image - # terratorch + # via scikit-image tiktoken==0.12.0 # via # -c requirements/common.txt @@ -1417,8 +1197,6 @@ timm==1.0.17 # -r requirements/test/rocm.in # open-clip-torch # segmentation-models-pytorch - # terratorch - # torchgeo tokenizers==0.22.2 # via # -c requirements/common.txt @@ -1429,16 +1207,6 @@ tomli==2.4.0 # via schemathesis tomli-w==1.2.0 # via schemathesis -torchgeo==0.7.0 - # via - # -r requirements/test/rocm.in - # terratorch -torchmetrics==1.9.0 - # via - # lightning - # pytorch-lightning - # terratorch - # torchgeo tqdm==4.67.3 # via # -r requirements/test/../common.txt @@ -1446,8 +1214,6 @@ tqdm==4.67.3 # evaluate # gguf # huggingface-hub - # lightly - # lightning # lm-eval # mteb # nltk @@ -1456,11 +1222,8 @@ tqdm==4.67.3 # optuna # peft # pqdm - # pytorch-lightning # segmentation-models-pytorch # sentence-transformers - # tacoreader - # terratorch # transformers transformers==5.5.3 # via @@ -1492,8 +1255,6 @@ typer==0.24.1 # huggingface-hub # perceptron # transformers -typeshed-client==2.9.0 - # via jsonargparse typing-extensions==4.15.0 # via # -c requirements/common.txt @@ -1511,8 +1272,6 @@ typing-extensions==4.15.0 # grpcio # huggingface-hub # librosa - # lightning - # lightning-utilities # lm-eval # mcp # mistral-common @@ -1527,18 +1286,14 @@ typing-extensions==4.15.0 # pydantic # pydantic-core # pydantic-extra-types - # pytorch-lightning # referencing # rich-toolkit # sentence-transformers # sqlalchemy # starlette # torch - # torchgeo # typeguard - # typeshed-client # typing-inspection - # wandb # xgrammar typing-inspection==0.4.2 # via @@ -1555,7 +1310,6 @@ urllib3==2.6.3 # blobfile # botocore # docker - # lightly # requests # responses # sentry-sdk @@ -1575,8 +1329,6 @@ virtualenv==21.2.0 # via ray vocos==0.1.0 # via -r requirements/test/rocm.in -wandb==0.25.1 - # via terratorch watchfiles==1.1.1 # via # -r requirements/test/../common.txt @@ -1588,15 +1340,11 @@ webcolors==25.10.0 websockets==16.0 # via uvicorn werkzeug==3.1.6 - # via - # schemathesis - # tensorboard + # via schemathesis word2number==1.1 # via lm-eval wrapt==2.1.2 # via smart-open -xarray==2026.2.0 - # via rioxarray xgrammar==0.1.33 # via # -c requirements/common.txt From aff00a19ef579fa95022429f04d1572f5993ca58 Mon Sep 17 00:00:00 2001 From: Stefano Castagnetta Date: Thu, 30 Apr 2026 16:29:53 +0200 Subject: [PATCH 2/2] [CI/Build] Skip terratorch tests cleanly while terratorch is unavailable Without terratorch installed, four test modules ERROR at collection or runtime instead of skipping cleanly: tests/models/test_terratorch.py tests/plugins_tests/test_terratorch_io_processor_plugins.py tests/models/multimodal/pooling/test_prithvi_mae.py tests/entrypoints/openai/chat_completion/test_vision_embeds.py Three are entirely Prithvi-scoped, so a module-level `pytest.importorskip("terratorch", reason=...)` is sufficient. The fourth (`test_vision_embeds.py`) also contains a Qwen3-VL test that must keep running; it gets a per-test `@pytest.mark.skipif` on the Prithvi case only. This unblocks every CI run that does not install terratorch (which is the case under #41376 / #41377). Once PyPI lifts the `lightning` quarantine and terratorch is restored to the lockfiles, these guards no-op (find_spec succeeds, importorskip does not skip). Signed-off-by: Stefano Castagnetta --- .../openai/chat_completion/test_vision_embeds.py | 11 +++++++++++ tests/models/multimodal/pooling/test_prithvi_mae.py | 5 +++++ tests/models/test_terratorch.py | 5 +++++ .../test_terratorch_io_processor_plugins.py | 5 +++++ 4 files changed, 26 insertions(+) diff --git a/tests/entrypoints/openai/chat_completion/test_vision_embeds.py b/tests/entrypoints/openai/chat_completion/test_vision_embeds.py index 574a8f1c86a9..da9787e3f89e 100644 --- a/tests/entrypoints/openai/chat_completion/test_vision_embeds.py +++ b/tests/entrypoints/openai/chat_completion/test_vision_embeds.py @@ -1,6 +1,8 @@ # SPDX-License-Identifier: Apache-2.0 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project +import importlib.util + import numpy as np import pybase64 as base64 import pytest @@ -10,7 +12,16 @@ from tests.utils import RemoteOpenAIServer from vllm.utils.serial_utils import tensor2base64 +# Prithvi requires terratorch, which is temporarily unavailable while PyPI has +# `lightning` quarantined (#41376). Skip just the Prithvi case; leave the +# Qwen3-VL case in the same file untouched. +_TERRATORCH_AVAILABLE = importlib.util.find_spec("terratorch") is not None + +@pytest.mark.skipif( + not _TERRATORCH_AVAILABLE, + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376", +) @pytest.mark.parametrize( "model_name", ["ibm-nasa-geospatial/Prithvi-EO-2.0-300M-TL-Sen1Floods11"] ) diff --git a/tests/models/multimodal/pooling/test_prithvi_mae.py b/tests/models/multimodal/pooling/test_prithvi_mae.py index 19154c27da9a..6340a07c83b2 100644 --- a/tests/models/multimodal/pooling/test_prithvi_mae.py +++ b/tests/models/multimodal/pooling/test_prithvi_mae.py @@ -6,6 +6,11 @@ from ....conftest import VllmRunner +pytest.importorskip( + "terratorch", + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376", +) + def _run_test( vllm_runner: type[VllmRunner], diff --git a/tests/models/test_terratorch.py b/tests/models/test_terratorch.py index 71125dbe94f8..fccc635a0dcf 100644 --- a/tests/models/test_terratorch.py +++ b/tests/models/test_terratorch.py @@ -7,6 +7,11 @@ from tests.conftest import VllmRunner from tests.utils import create_new_process_for_each_test +pytest.importorskip( + "terratorch", + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376", +) + @create_new_process_for_each_test() # Hangs otherwise @pytest.mark.parametrize( diff --git a/tests/plugins_tests/test_terratorch_io_processor_plugins.py b/tests/plugins_tests/test_terratorch_io_processor_plugins.py index 34799b3c42c0..01e120a24406 100644 --- a/tests/plugins_tests/test_terratorch_io_processor_plugins.py +++ b/tests/plugins_tests/test_terratorch_io_processor_plugins.py @@ -11,6 +11,11 @@ from tests.utils import RemoteOpenAIServer from vllm.entrypoints.pooling.pooling.protocol import IOProcessorResponse +pytest.importorskip( + "terratorch", + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376", +) + models_config = { "ibm-nasa-geospatial/Prithvi-EO-2.0-300M-TL-Sen1Floods11": { "image_url": "https://huggingface.co/christian-pinto/Prithvi-EO-2.0-300M-TL-VLLM/resolve/main/valencia_example_2024-10-26.tiff", # noqa: E501