@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / 10558f5f46
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 4, 11:15 UTC

[CI/Build] Skip terratorch + torchgeo while PyPI has lightning quarantined (#41377)

Signed-off-by: Stefano Castagnetta <scastagnetta@nvidia.com>

Stefano Castagnetta committed Apr 30, 2026 at 16:59 UTC 10558f5f4608f825d06018def1317e7d3a96d6fe
8 files changed +63 -542
requirements/test/cuda.in
+5 -1
@@ -61,7 +61,11 @@ fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
61 instanttensor>=0.1.5
62 pydantic>=2.12 # 2.11 leads to error on python 3.13
63 decord==0.6.0; platform_machine == "x86_64"
64 -terratorch >= 1.2.2 # Required for Prithvi tests
64 +# terratorch is temporarily disabled while PyPI has the `lightning` package
65 +# in `quarantined` status (every published terratorch version transitively
66 +# requires `lightning`, so the resolver fails with "no versions of lightning").
67 +# Re-enable once PyPI lifts the quarantine. Tracked in #41376.
68 +# terratorch >= 1.2.2 # Required for Prithvi tests
69 imagehash # Required for Prithvi tests
70 segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests
71
requirements/test/cuda.txt
+11 -274
@@ -1,15 +1,9 @@
1 # This file was autogenerated by uv via the following command:
2 # 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
3 absl-py==2.1.0
4 - # via
5 - # rouge-score
6 - # tensorboard
4 + # via rouge-score
5 accelerate==1.13.0
6 # via peft
9 -aenum==3.1.16
10 - # via lightly
11 -affine==2.4.0
12 - # via rasterio
7 aiohappyeyeballs==2.6.1
8 # via aiohttp
9 aiohttp==3.13.3
@@ -25,22 +19,14 @@ aiohttp-cors==0.8.1
19 # via ray
20 aiosignal==1.4.0
21 # via aiohttp
28 -albucore==0.0.16
29 - # via terratorch
22 albumentations==1.4.6
31 - # via
32 - # -r requirements/test/cuda.in
33 - # terratorch
23 + # via -r requirements/test/cuda.in
24 alembic==1.16.4
25 # via optuna
26 annotated-doc==0.0.4
27 # via fastapi
28 annotated-types==0.7.0
29 # via pydantic
40 -antlr4-python3-runtime==4.9.3
41 - # via
42 - # hydra-core
43 - # omegaconf
30 anyio==4.6.2.post1
31 # via
32 # httpx
@@ -54,12 +40,10 @@ arrow==1.3.0
40 attrs==24.2.0
41 # via
42 # aiohttp
57 - # fiona
43 # hypothesis
44 # jsonlines
45 # jsonschema
46 # pytest-subtests
62 - # rasterio
47 # referencing
48 audioread==3.0.1
49 # via librosa
@@ -78,9 +62,7 @@ backoff==2.2.1
62 # -r requirements/test/cuda.in
63 # schemathesis
64 bitsandbytes==0.49.2
81 - # via
82 - # -r requirements/test/cuda.in
83 - # lightning
65 + # via -r requirements/test/cuda.in
66 black==24.10.0
67 # via datamodel-code-generator
68 blobfile==3.0.0
@@ -103,15 +85,9 @@ cachetools==5.5.2
85 # via google-auth
86 certifi==2024.8.30
87 # via
106 - # fiona
88 # httpcore
89 # httpx
109 - # lightly
110 - # pyogrio
111 - # pyproj
112 - # rasterio
90 # requests
114 - # sentry-sdk
91 cffi==2.0.0
92 # via
93 # cryptography
@@ -125,25 +101,12 @@ chz==0.3.0
101 click==8.1.7
102 # via
103 # black
128 - # click-plugins
129 - # cligj
130 - # fiona
104 # jiwer
105 # nltk
133 - # rasterio
106 # ray
107 # schemathesis
108 # typer
109 # uvicorn
138 - # wandb
139 -click-plugins==1.1.1.2
140 - # via
141 - # fiona
142 - # rasterio
143 -cligj==0.7.2
144 - # via
145 - # fiona
146 - # rasterio
110 colorama==0.4.6
111 # via
112 # perceptron
@@ -191,8 +154,6 @@ decorator==5.1.1
154 # via librosa
155 decord==0.6.0
156 # via -r requirements/test/cuda.in
194 -diffusers==0.36.0
195 - # via terratorch
157 dill==0.3.8
158 # via
159 # datasets
@@ -207,14 +168,10 @@ docker==7.1.0
168 # via gpt-oss
169 docopt==0.6.2
170 # via num2words
210 -docstring-parser==0.17.0
211 - # via jsonargparse
171 einops==0.8.1
172 # via
173 # -r requirements/test/cuda.in
174 # encodec
216 - # terratorch
217 - # torchgeo
175 # vector-quantize-pytorch
176 # vocos
177 einx==0.3.0
@@ -244,13 +201,10 @@ filelock==3.16.1
201 # -c requirements/common.txt
202 # blobfile
203 # datasets
247 - # diffusers
204 # huggingface-hub
205 # ray
206 # torch
207 # virtualenv
252 -fiona==1.10.1
253 - # via torchgeo
208 fonttools==4.55.0
209 # via matplotlib
210 fqdn==1.5.1
@@ -267,9 +221,6 @@ fsspec==2024.12.0
221 # evaluate
222 # fastparquet
223 # huggingface-hub
270 - # lightning
271 - # pytorch-lightning
272 - # tacoreader
224 # torch
225 ftfy==6.3.1
226 # via open-clip-torch
@@ -277,12 +228,6 @@ genai-perf==0.0.16
228 # via -r requirements/test/cuda.in
229 genson==1.3.0
230 # via datamodel-code-generator
280 -geopandas==1.0.1
281 - # via terratorch
282 -gitdb==4.0.12
283 - # via gitpython
284 -gitpython==3.1.44
285 - # via wandb
231 google-api-core==2.24.2
232 # via
233 # google-cloud-core
@@ -317,7 +262,6 @@ grpcio==1.78.0
262 # -r requirements/test/cuda.in
263 # grpcio-reflection
264 # ray
320 - # tensorboard
265 grpcio-reflection==1.78.0
266 # via -r requirements/test/cuda.in
267 h11==0.14.0
@@ -326,8 +270,6 @@ h11==0.14.0
270 # uvicorn
271 h2==4.3.0
272 # via httpx
329 -h5py==3.13.0
330 - # via terratorch
273 harfile==0.3.0
274 # via schemathesis
275 hf-xet==1.4.3
@@ -343,7 +285,6 @@ httpcore==1.0.6
285 httpx==0.27.2
286 # via
287 # -r requirements/test/cuda.in
346 - # diffusers
288 # huggingface-hub
289 # perceptron
290 # schemathesis
@@ -351,23 +292,17 @@ huggingface-hub==1.10.2
292 # via
293 # accelerate
294 # datasets
354 - # diffusers
295 # evaluate
296 # open-clip-torch
297 # peft
298 # segmentation-models-pytorch
299 # sentence-transformers
360 - # terratorch
300 # timm
301 # tokenizers
302 # transformers
303 # vocos
304 humanize==4.11.0
305 # via runai-model-streamer
367 -hydra-core==1.3.2
368 - # via
369 - # lightly
370 - # lightning
306 hyperframe==6.1.0
307 # via h2
308 hypothesis==6.131.0
@@ -392,11 +327,7 @@ imagehash==4.3.2
327 imageio==2.37.0
328 # via scikit-image
329 importlib-metadata==8.7.0
395 - # via
396 - # diffusers
397 - # opentelemetry-api
398 -importlib-resources==6.5.2
399 - # via typeshed-client
330 + # via opentelemetry-api
331 inflect==5.6.2
332 # via datamodel-code-generator
333 iniconfig==2.0.0
@@ -426,14 +357,8 @@ joblib==1.4.2
357 # librosa
358 # nltk
359 # scikit-learn
429 -jsonargparse==4.46.0
430 - # via
431 - # lightning
432 - # terratorch
360 jsonlines==4.0.0
361 # via lm-eval
435 -jsonnet==0.21.0
436 - # via jsonargparse
362 jsonpointer==3.0.0
363 # via jsonschema
364 jsonschema==4.23.0
@@ -452,10 +377,6 @@ kaleido==0.2.1
377 # via genai-perf
378 kiwisolver==1.4.7
379 # via matplotlib
455 -kornia==0.8.1
456 - # via torchgeo
457 -kornia-rs==0.1.9
458 - # via kornia
380 lazy-loader==0.4
381 # via
382 # librosa
@@ -464,21 +385,6 @@ libnacl==2.1.0
385 # via tensorizer
386 librosa==0.10.2.post1
387 # via -r requirements/test/cuda.in
467 -lightly==1.5.22
468 - # via
469 - # terratorch
470 - # torchgeo
471 -lightly-utils==0.0.2
472 - # via lightly
473 -lightning==2.6.1
474 - # via
475 - # terratorch
476 - # torchgeo
477 -lightning-utilities==0.14.3
478 - # via
479 - # lightning
480 - # pytorch-lightning
481 - # torchmetrics
388 llvmlite==0.47.0
389 # via numba
390 lm-eval==0.4.11
@@ -490,8 +396,6 @@ lxml==5.3.0
396 # sacrebleu
397 mako==1.3.10
398 # via alembic
493 -markdown==3.8.2
494 - # via tensorboard
399 markdown-it-py==3.0.0
400 # via rich
401 markupsafe==3.0.1
@@ -500,11 +404,7 @@ markupsafe==3.0.1
404 # mako
405 # werkzeug
406 matplotlib==3.9.2
503 - # via
504 - # -r requirements/test/cuda.in
505 - # lightning
506 - # pycocotools
507 - # torchgeo
407 + # via -r requirements/test/cuda.in
408 mbstrdecoder==1.1.3
409 # via
410 # dataproperty
@@ -559,7 +459,6 @@ numpy==2.2.6
459 # via
460 # -r requirements/test/cuda.in
461 # accelerate
562 - # albucore
462 # albumentations
463 # bitsandbytes
464 # bm25s
@@ -567,19 +466,14 @@ numpy==2.2.6
466 # cupy-cuda12x
467 # datasets
468 # decord
570 - # diffusers
469 # einx
470 # encodec
471 # evaluate
472 # fastparquet
473 # genai-perf
576 - # geopandas
577 - # h5py
474 # imagehash
475 # imageio
476 # librosa
581 - # lightly
582 - # lightly-utils
477 # lm-eval
478 # matplotlib
479 # mistral-common
@@ -591,11 +485,7 @@ numpy==2.2.6
485 # patsy
486 # peft
487 # perceptron
594 - # pycocotools
595 - # pyogrio
488 # pywavelets
597 - # rasterio
598 - # rioxarray
489 # rouge-score
490 # runai-model-streamer
491 # sacrebleu
@@ -603,21 +493,14 @@ numpy==2.2.6
493 # scikit-learn
494 # scipy
495 # segmentation-models-pytorch
606 - # shapely
496 # soxr
497 # statsmodels
609 - # tensorboard
610 - # tensorboardx
498 # tensorizer
612 - # terratorch
499 # tifffile
614 - # torchgeo
615 - # torchmetrics
500 # torchvision
501 # transformers
502 # tritonclient
503 # vocos
620 - # xarray
504 nvidia-cublas==13.1.0.3
505 # via
506 # cuda-toolkit
@@ -657,10 +540,6 @@ nvidia-nvshmem-cu13==3.4.5
540 # via torch
541 nvidia-nvtx==13.0.85
542 # via cuda-toolkit
660 -omegaconf==2.3.0
661 - # via
662 - # hydra-core
663 - # lightning
543 open-clip-torch==2.32.0
544 # via -r requirements/test/cuda.in
545 openai-harmony==0.0.4
@@ -675,7 +554,6 @@ opencv-python-headless==4.13.0.90
554 # via
555 # -c requirements/common.txt
556 # -r requirements/test/cuda.in
678 - # albucore
557 # albumentations
558 # mistral-common
559 openpyxl==3.1.5
@@ -710,44 +588,27 @@ packaging==24.2
588 # datasets
589 # evaluate
590 # fastparquet
713 - # geopandas
591 # huggingface-hub
715 - # hydra-core
716 - # kornia
592 # lazy-loader
718 - # lightning
719 - # lightning-utilities
593 # matplotlib
594 # optuna
595 # peft
596 # plotly
597 # pooch
725 - # pyogrio
598 # pytest
599 # pytest-rerunfailures
728 - # pytorch-lightning
600 # ray
730 - # rioxarray
601 # scikit-image
602 # statsmodels
733 - # tensorboard
734 - # tensorboardx
735 - # torchmetrics
603 # transformers
604 # typepy
738 - # wandb
739 - # xarray
605 pandas==2.2.3
606 # via
607 # datasets
608 # evaluate
609 # fastparquet
610 # genai-perf
746 - # geopandas
611 # statsmodels
748 - # tacoreader
749 - # torchgeo
750 - # xarray
612 pathspec==0.12.1
613 # via black
614 pathvalidate==3.2.1
@@ -762,25 +623,20 @@ perf-analyzer==0.1.0
623 # via genai-perf
624 pillow==10.4.0
625 # via
765 - # diffusers
626 # genai-perf
627 # imagehash
628 # imageio
769 - # lightly-utils
629 # matplotlib
630 # mistral-common
631 # perceptron
632 # scikit-image
633 # segmentation-models-pytorch
775 - # tensorboard
776 - # torchgeo
634 # torchvision
635 platformdirs==4.3.6
636 # via
637 # black
638 # pooch
639 # virtualenv
783 - # wandb
640 plotly==5.24.1
641 # via
642 # -r requirements/test/cuda.in
@@ -817,10 +673,7 @@ protobuf==6.33.6
673 # opentelemetry-proto
674 # proto-plus
675 # ray
820 - # tensorboard
821 - # tensorboardx
676 # tensorizer
823 - # wandb
677 psutil==6.1.0
678 # via
679 # accelerate
@@ -834,16 +687,12 @@ pyarrow==23.0.0
687 # via
688 # datasets
689 # genai-perf
837 - # tacoreader
838 - # terratorch
690 pyasn1==0.6.1
691 # via
692 # pyasn1-modules
693 # rsa
694 pyasn1-modules==0.4.2
695 # via google-auth
845 -pycocotools==2.0.8
846 - # via terratorch
696 pycountry==24.6.1
697 # via pydantic-extra-types
698 pycparser==2.22
@@ -858,13 +707,11 @@ pydantic==2.12.0
707 # datamodel-code-generator
708 # fastapi
709 # gpt-oss
861 - # lightly
710 # mistral-common
711 # mteb
712 # openai-harmony
713 # pydantic-extra-types
714 # ray
867 - # wandb
715 pydantic-core==2.41.1
716 # via pydantic
717 pydantic-extra-types==2.10.5
@@ -873,17 +720,8 @@ pygments==2.18.0
720 # via rich
721 pyjwt==2.11.0
722 # via msal
876 -pyogrio==0.11.0
877 - # via geopandas
723 pyparsing==3.2.0
879 - # via
880 - # matplotlib
881 - # rasterio
882 -pyproj==3.7.1
883 - # via
884 - # geopandas
885 - # rioxarray
886 - # torchgeo
724 + # via matplotlib
725 pyrate-limiter==3.7.0
726 # via schemathesis
727 pystemmer==3.0.0
@@ -920,22 +758,15 @@ pytest-subtests==0.14.1
758 # via schemathesis
759 pytest-timeout==2.3.1
760 # via -r requirements/test/cuda.in
923 -python-box==7.3.2
924 - # via terratorch
761 python-dateutil==2.9.0.post0
762 # via
763 # arrow
764 # botocore
929 - # lightly
765 # matplotlib
766 # pandas
767 # typepy
768 python-rapidjson==1.20
769 # via tritonclient
935 -pytorch-lightning==2.5.2
936 - # via
937 - # lightly
938 - # lightning
770 pytrec-eval-terrier==0.5.7
771 # via mteb
772 pytz==2024.2
@@ -952,26 +783,16 @@ pyyaml==6.0.2
783 # datasets
784 # genai-perf
785 # huggingface-hub
955 - # jsonargparse
956 - # lightning
957 - # omegaconf
786 # optuna
787 # peft
960 - # pytorch-lightning
788 # ray
789 # responses
790 # schemathesis
791 # timm
792 # transformers
793 # vocos
967 - # wandb
794 rapidfuzz==3.12.1
795 # via jiwer
970 -rasterio==1.4.3
971 - # via
972 - # rioxarray
973 - # terratorch
974 - # torchgeo
796 ray==2.48.0
797 # via -r requirements/test/cuda.in
798 redis==5.2.0
@@ -982,7 +803,6 @@ referencing==0.35.1
803 # jsonschema-specifications
804 regex==2026.2.28
805 # via
985 - # diffusers
806 # nltk
807 # open-clip-torch
808 # sacrebleu
@@ -994,13 +814,11 @@ requests==2.32.3
814 # azure-core
815 # buildkite-test-collector
816 # datasets
997 - # diffusers
817 # docker
818 # evaluate
819 # google-api-core
820 # google-cloud-storage
821 # gpt-oss
1003 - # lightly
822 # lm-eval
823 # mistral-common
824 # msal
@@ -1010,9 +828,7 @@ requests==2.32.3
828 # responses
829 # schemathesis
830 # starlette-testclient
1013 - # tacoreader
831 # tiktoken
1015 - # wandb
832 responses==0.25.3
833 # via genai-perf
834 rfc3339-validator==0.1.4
@@ -1022,13 +838,9 @@ rfc3987==1.3.8
838 rich==13.9.4
839 # via
840 # genai-perf
1025 - # lightning
841 # mteb
842 # perceptron
1028 - # terratorch
843 # typer
1030 -rioxarray==0.19.0
1031 - # via terratorch
844 rouge-score==0.1.2
845 # via lm-eval
846 rpds-py==0.20.1
@@ -1037,8 +849,6 @@ rpds-py==0.20.1
849 # referencing
850 rsa==4.9.1
851 # via google-auth
1040 -rtree==1.4.0
1041 - # via torchgeo
852 runai-model-streamer==0.15.7
853 # via -r requirements/test/cuda.in
854 runai-model-streamer-azure==0.15.7
@@ -1054,7 +864,6 @@ sacrebleu==2.4.3
864 safetensors==0.4.5
865 # via
866 # accelerate
1057 - # diffusers
867 # open-clip-torch
868 # peft
869 # segmentation-models-pytorch
@@ -1063,9 +872,7 @@ safetensors==0.4.5
872 schemathesis==3.39.15
873 # via -r requirements/test/cuda.in
874 scikit-image==0.25.2
1066 - # via
1067 - # albumentations
1068 - # terratorch
875 + # via albumentations
876 scikit-learn==1.5.2
877 # via
878 # albumentations
@@ -1073,7 +880,6 @@ scikit-learn==1.5.2
880 # lm-eval
881 # mteb
882 # sentence-transformers
1076 - # terratorch
883 scipy==1.13.1
884 # via
885 # albumentations
@@ -1087,27 +893,16 @@ scipy==1.13.1
893 # statsmodels
894 # vocos
895 segmentation-models-pytorch==0.5.0
1090 - # via
1091 - # -r requirements/test/cuda.in
1092 - # terratorch
1093 - # torchgeo
896 + # via -r requirements/test/cuda.in
897 sentence-transformers==5.2.0
898 # via
899 # -r requirements/test/cuda.in
900 # mteb
1098 -sentry-sdk==2.52.0
1099 - # via wandb
901 setuptools==77.0.3
902 # via
903 # -c requirements/common.txt
1103 - # lightning-utilities
904 # pytablewriter
1105 - # tensorboard
905 # torch
1107 -shapely==2.1.1
1108 - # via
1109 - # geopandas
1110 - # torchgeo
906 shellingham==1.5.4
907 # via
908 # perceptron
@@ -1116,15 +911,12 @@ six==1.16.0
911 # via
912 # -c requirements/common.txt
913 # junit-xml
1119 - # lightly
914 # opencensus
915 # python-dateutil
916 # rfc3339-validator
917 # rouge-score
918 smart-open==7.1.0
919 # via ray
1126 -smmap==5.0.2
1127 - # via gitdb
920 sniffio==1.3.1
921 # via
922 # anyio
@@ -1166,8 +958,6 @@ tabledata==1.3.3
958 # via pytablewriter
959 tabulate==0.9.0
960 # via sacrebleu
1169 -tacoreader==0.5.6
1170 - # via terratorch
961 tblib==3.1.0
962 # via -r requirements/test/cuda.in
963 tcolorpy==0.1.6
@@ -1177,26 +967,14 @@ tenacity==9.1.2
967 # gpt-oss
968 # lm-eval
969 # plotly
1180 -tensorboard==2.20.0
1181 - # via terratorch
1182 -tensorboard-data-server==0.7.2
1183 - # via tensorboard
1184 -tensorboardx==2.6.4
1185 - # via lightning
970 tensorizer==2.10.1
971 # via -r requirements/test/cuda.in
972 termcolor==3.1.0
1189 - # via
1190 - # gpt-oss
1191 - # terratorch
1192 -terratorch==1.2.2
1193 - # via -r requirements/test/cuda.in
973 + # via gpt-oss
974 threadpoolctl==3.5.0
975 # via scikit-learn
976 tifffile==2025.3.30
1197 - # via
1198 - # scikit-image
1199 - # terratorch
977 + # via scikit-image
978 tiktoken==0.12.0
979 # via
980 # -c requirements/common.txt
@@ -1208,8 +986,6 @@ timm==1.0.17
986 # -r requirements/test/cuda.in
987 # open-clip-torch
988 # segmentation-models-pytorch
1211 - # terratorch
1212 - # torchgeo
989 tokenizers==0.22.2
990 # via
991 # -c requirements/common.txt
@@ -1227,21 +1003,14 @@ torch==2.11.0+cu130
1003 # bitsandbytes
1004 # encodec
1005 # instanttensor
1230 - # kornia
1231 - # lightly
1232 - # lightning
1006 # mteb
1007 # open-clip-torch
1008 # peft
1236 - # pytorch-lightning
1009 # runai-model-streamer
1010 # segmentation-models-pytorch
1011 # sentence-transformers
1012 # tensorizer
1241 - # terratorch
1013 # timm
1243 - # torchgeo
1244 - # torchmetrics
1014 # torchvision
1015 # vector-quantize-pytorch
1016 # vocos
@@ -1251,31 +1020,18 @@ torchaudio==2.11.0+cu130
1020 # -r requirements/test/cuda.in
1021 # encodec
1022 # vocos
1254 -torchgeo==0.7.0
1255 - # via terratorch
1256 -torchmetrics==1.7.4
1257 - # via
1258 - # lightning
1259 - # pytorch-lightning
1260 - # terratorch
1261 - # torchgeo
1023 torchvision==0.26.0+cu130
1024 # via
1025 # -c requirements/cuda.txt
1026 # -r requirements/test/cuda.in
1266 - # lightly
1027 # open-clip-torch
1028 # segmentation-models-pytorch
1269 - # terratorch
1029 # timm
1271 - # torchgeo
1030 tqdm==4.67.3
1031 # via
1032 # datasets
1033 # evaluate
1034 # huggingface-hub
1277 - # lightly
1278 - # lightning
1035 # lm-eval
1036 # mteb
1037 # nltk
@@ -1283,11 +1039,8 @@ tqdm==4.67.3
1039 # optuna
1040 # peft
1041 # pqdm
1286 - # pytorch-lightning
1042 # segmentation-models-pytorch
1043 # sentence-transformers
1289 - # tacoreader
1290 - # terratorch
1044 # transformers
1045 transformers==5.5.3
1046 # via
@@ -1316,8 +1069,6 @@ typer==0.15.2
1069 # transformers
1070 types-python-dateutil==2.9.0.20241206
1071 # via arrow
1319 -typeshed-client==2.8.2
1320 - # via jsonargparse
1072 typing-extensions==4.15.0
1073 # via
1074 # -c requirements/common.txt
@@ -1332,8 +1083,6 @@ typing-extensions==4.15.0
1083 # grpcio
1084 # huggingface-hub
1085 # librosa
1335 - # lightning
1336 - # lightning-utilities
1086 # lm-eval
1087 # mistral-common
1088 # mteb
@@ -1344,16 +1093,12 @@ typing-extensions==4.15.0
1093 # pydantic
1094 # pydantic-core
1095 # pydantic-extra-types
1347 - # pytorch-lightning
1096 # sentence-transformers
1097 # sqlalchemy
1098 # starlette
1099 # torch
1352 - # torchgeo
1100 # typer
1354 - # typeshed-client
1101 # typing-inspection
1356 - # wandb
1102 typing-inspection==0.4.2
1103 # via pydantic
1104 tzdata==2024.2
@@ -1365,10 +1110,8 @@ urllib3==2.2.3
1110 # blobfile
1111 # botocore
1112 # docker
1368 - # lightly
1113 # requests
1114 # responses
1371 - # sentry-sdk
1115 # tritonclient
1116 uvicorn==0.35.0
1117 # via gpt-oss
@@ -1378,22 +1121,16 @@ virtualenv==20.31.2
1121 # via ray
1122 vocos==0.1.0
1123 # via -r requirements/test/cuda.in
1381 -wandb==0.24.2
1382 - # via terratorch
1124 wcwidth==0.2.13
1125 # via ftfy
1126 webcolors==24.11.1
1127 # via jsonschema
1128 werkzeug==3.1.3
1388 - # via
1389 - # schemathesis
1390 - # tensorboard
1129 + # via schemathesis
1130 word2number==1.1
1131 # via lm-eval
1132 wrapt==1.17.2
1133 # via smart-open
1395 -xarray==2025.7.1
1396 - # via rioxarray
1134 xxhash==3.5.0
1135 # via
1136 # datasets
requirements/test/rocm.in
+8 -2
@@ -61,7 +61,11 @@ pydantic>=2.12 # 2.11 leads to error on python 3.13
61 decord==0.6.0
62
63 # Prithvi tests
64 -terratorch>=1.2.2
64 +# terratorch is temporarily disabled while PyPI has the `lightning` package
65 +# in `quarantined` status (every published terratorch version transitively
66 +# requires `lightning`, so the resolver fails with "no versions of lightning").
67 +# Re-enable once PyPI lifts the quarantine. Tracked in #41376.
68 +# terratorch>=1.2.2
69 imagehash # Required for Prithvi tests
70 segmentation-models-pytorch>0.4.0 # Required for Prithvi tests
71
@@ -79,5 +83,7 @@ plotly # required for perf comparison html report
83
84 # ROCm-specific extras (not in CUDA cuda.in)
85 rapidfuzz
82 -torchgeo==0.7.0
86 +# torchgeo also pulled in `lightning` transitively; disabled for the same
87 +# quarantine reason as terratorch above. Restore once the quarantine clears.
88 +# torchgeo==0.7.0
89 multiprocess==0.70.16
requirements/test/rocm.txt
+13 -265
@@ -1,15 +1,9 @@
1 # This file was autogenerated by uv via the following command:
2 # 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
3 absl-py==2.4.0
4 - # via
5 - # rouge-score
6 - # tensorboard
4 + # via rouge-score
5 accelerate==1.13.0
6 # via peft
9 -aenum==3.1.17
10 - # via lightly
11 -affine==2.4.0
12 - # via rasterio
7 aiohappyeyeballs==2.6.1
8 # via aiohttp
9 aiohttp==3.13.3
@@ -25,12 +19,8 @@ aiohttp-cors==0.8.1
19 # via ray
20 aiosignal==1.4.0
21 # via aiohttp
28 -albucore==0.1.2
29 - # via terratorch
22 albumentations==1.4.6
31 - # via
32 - # -r requirements/test/rocm.in
33 - # terratorch
23 + # via -r requirements/test/rocm.in
24 alembic==1.18.4
25 # via optuna
26 annotated-doc==0.0.4
@@ -43,10 +33,6 @@ anthropic==0.93.0
33 # via
34 # -c requirements/common.txt
35 # -r requirements/test/../common.txt
46 -antlr4-python3-runtime==4.9.3
47 - # via
48 - # hydra-core
49 - # omegaconf
36 anyio==4.13.0
37 # via
38 # anthropic
@@ -67,11 +53,9 @@ astor==0.8.1
53 attrs==26.1.0
54 # via
55 # aiohttp
70 - # fiona
56 # jsonlines
57 # jsonschema
58 # pytest-subtests
74 - # rasterio
59 # referencing
60 audioread==3.0.1
61 # via librosa
@@ -90,9 +74,7 @@ backoff==2.2.1
74 # -r requirements/test/rocm.in
75 # schemathesis
76 bitsandbytes==0.49.2
93 - # via
94 - # -r requirements/test/rocm.in
95 - # lightning
77 + # via -r requirements/test/rocm.in
78 black==26.3.1
79 # via datamodel-code-generator
80 blake3==1.0.8
@@ -119,13 +101,8 @@ cbor2==5.9.0
101 # via -r requirements/test/../common.txt
102 certifi==2026.2.25
103 # via
122 - # fiona
104 # httpcore
105 # httpx
125 - # lightly
126 - # pyogrio
127 - # pyproj
128 - # rasterio
106 # requests
107 # sentry-sdk
108 cffi==1.17.1
@@ -143,24 +120,13 @@ chz==0.4.0
120 click==8.3.1
121 # via
122 # black
146 - # click-plugins
147 - # cligj
148 - # fiona
123 # jiwer
124 # nltk
151 - # rasterio
125 # ray
126 # rich-toolkit
127 # schemathesis
128 # typer
129 # uvicorn
157 - # wandb
158 -click-plugins==1.1.1.2
159 - # via fiona
160 -cligj==0.7.2
161 - # via
162 - # fiona
163 - # rasterio
130 cloudpickle==3.1.2
131 # via -r requirements/test/../common.txt
132 colorama==0.4.6
@@ -211,8 +177,6 @@ depyf==0.20.0
177 # via
178 # -c requirements/common.txt
179 # -r requirements/test/../common.txt
214 -diffusers==0.37.0
215 - # via terratorch
180 dill==0.3.8
181 # via
182 # datasets
@@ -237,16 +201,12 @@ docker==7.1.0
201 docopt==0.6.2
202 # via num2words
203 docstring-parser==0.17.0
240 - # via
241 - # anthropic
242 - # jsonargparse
204 + # via anthropic
205 einops==0.8.2
206 # via
207 # -r requirements/test/../common.txt
208 # -r requirements/test/rocm.in
209 # encodec
248 - # terratorch
249 - # torchgeo
210 # vector-quantize-pytorch
211 # vocos
212 einx==0.4.2
@@ -283,14 +243,11 @@ filelock==3.25.2
243 # -r requirements/test/../common.txt
244 # blobfile
245 # datasets
286 - # diffusers
246 # huggingface-hub
247 # python-discovery
248 # ray
249 # torch
250 # virtualenv
292 -fiona==1.10.1
293 - # via torchgeo
251 fonttools==4.62.1
252 # via matplotlib
253 fqdn==1.5.1
@@ -307,9 +264,6 @@ fsspec==2025.3.0
264 # evaluate
265 # fastparquet
266 # huggingface-hub
310 - # lightning
311 - # pytorch-lightning
312 - # tacoreader
267 # torch
268 ftfy==6.3.1
269 # via open-clip-torch
@@ -317,16 +271,10 @@ genai-perf==0.0.16
271 # via -r requirements/test/rocm.in
272 genson==1.3.0
273 # via datamodel-code-generator
320 -geopandas==1.1.3
321 - # via terratorch
274 gguf==0.18.0
275 # via
276 # -c requirements/common.txt
277 # -r requirements/test/../common.txt
326 -gitdb==4.0.12
327 - # via gitpython
328 -gitpython==3.1.46
329 - # via wandb
278 google-api-core==2.30.0
279 # via
280 # google-cloud-core
@@ -366,7 +314,6 @@ grpcio==1.78.0
314 # grpcio-reflection
315 # opentelemetry-exporter-otlp-proto-grpc
316 # ray
369 - # tensorboard
317 grpcio-reflection==1.78.0
318 # via
319 # -c requirements/rocm.txt
@@ -377,8 +324,6 @@ h11==0.16.0
324 # uvicorn
325 h2==4.3.0
326 # via httpx
380 -h5py==3.16.0
381 - # via terratorch
327 harfile==0.4.0
328 # via schemathesis
329 hf-xet==1.4.3
@@ -397,7 +342,6 @@ httpx==0.27.2
342 # via
343 # -r requirements/test/rocm.in
344 # anthropic
400 - # diffusers
345 # fastapi
346 # fastapi-cloud-cli
347 # huggingface-hub
@@ -412,23 +356,17 @@ huggingface-hub==1.10.2
356 # via
357 # accelerate
358 # datasets
415 - # diffusers
359 # evaluate
360 # open-clip-torch
361 # peft
362 # segmentation-models-pytorch
363 # sentence-transformers
421 - # terratorch
364 # timm
365 # tokenizers
366 # transformers
367 # vocos
368 humanize==4.15.0
369 # via runai-model-streamer
428 -hydra-core==1.3.2
429 - # via
430 - # lightly
431 - # lightning
370 hyperframe==6.1.0
371 # via h2
372 hypothesis==6.151.9
@@ -455,11 +393,7 @@ imagehash==4.3.2
393 imageio==2.37.3
394 # via scikit-image
395 importlib-metadata==8.7.1
458 - # via
459 - # diffusers
460 - # opentelemetry-api
461 -importlib-resources==6.5.2
462 - # via typeshed-client
396 + # via opentelemetry-api
397 inflect==7.5.0
398 # via datamodel-code-generator
399 iniconfig==2.3.0
@@ -497,14 +431,8 @@ joblib==1.5.3
431 # librosa
432 # nltk
433 # scikit-learn
500 -jsonargparse==4.47.0
501 - # via
502 - # lightning
503 - # terratorch
434 jsonlines==4.0.0
435 # via lm-eval
506 -jsonnet==0.21.0
507 - # via jsonargparse
436 jsonpointer==3.1.0
437 # via jsonschema
438 jsonschema==4.26.0
@@ -524,10 +452,6 @@ kaleido==1.0.0
452 # via genai-perf
453 kiwisolver==1.5.0
454 # via matplotlib
527 -kornia==0.8.2
528 - # via torchgeo
529 -kornia-rs==0.1.10
530 - # via kornia
455 lark==1.2.2
456 # via
457 # -c requirements/common.txt
@@ -540,21 +464,6 @@ libnacl==2.1.0
464 # via tensorizer
465 librosa==0.10.2.post1
466 # via -r requirements/test/rocm.in
543 -lightly==1.5.22
544 - # via
545 - # terratorch
546 - # torchgeo
547 -lightly-utils==0.0.2
548 - # via lightly
549 -lightning==2.6.1
550 - # via
551 - # terratorch
552 - # torchgeo
553 -lightning-utilities==0.15.3
554 - # via
555 - # lightning
556 - # pytorch-lightning
557 - # torchmetrics
467 llguidance==1.3.0
468 # via
469 # -c requirements/common.txt
@@ -580,8 +489,6 @@ lxml==6.0.2
489 # sacrebleu
490 mako==1.3.10
491 # via alembic
583 -markdown==3.10.2
584 - # via tensorboard
492 markdown-it-py==4.0.0
493 # via rich
494 markupsafe==3.0.3
@@ -590,10 +497,7 @@ markupsafe==3.0.3
497 # mako
498 # werkzeug
499 matplotlib==3.10.8
593 - # via
594 - # -r requirements/test/rocm.in
595 - # lightning
596 - # torchgeo
500 + # via -r requirements/test/rocm.in
501 mbstrdecoder==1.1.4
502 # via
503 # dataproperty
@@ -660,14 +564,11 @@ numba==0.65.0
564 # -c requirements/rocm.txt
565 # -r requirements/test/rocm.in
566 # librosa
663 -numkong==7.1.1
664 - # via albucore
567 numpy==2.2.6
568 # via
569 # -r requirements/test/../common.txt
570 # -r requirements/test/rocm.in
571 # accelerate
670 - # albucore
572 # albumentations
573 # bitsandbytes
574 # bm25s
@@ -675,20 +576,15 @@ numpy==2.2.6
576 # cupy-cuda12x
577 # datasets
578 # decord
678 - # diffusers
579 # einx
580 # encodec
581 # evaluate
582 # fastparquet
583 # genai-perf
684 - # geopandas
584 # gguf
686 - # h5py
585 # imagehash
586 # imageio
587 # librosa
690 - # lightly
691 - # lightly-utils
588 # lm-eval
589 # matplotlib
590 # mistral-common
@@ -700,12 +596,8 @@ numpy==2.2.6
596 # patsy
597 # peft
598 # perceptron
703 - # pycocotools
704 - # pyogrio
599 # pytrec-eval-terrier
600 # pywavelets
707 - # rasterio
708 - # rioxarray
601 # rouge-score
602 # runai-model-streamer
603 # sacrebleu
@@ -714,27 +606,16 @@ numpy==2.2.6
606 # scipy
607 # segmentation-models-pytorch
608 # sentence-transformers
717 - # shapely
609 # soundfile
610 # soxr
611 # statsmodels
721 - # tensorboard
722 - # tensorboardx
612 # tensorizer
724 - # terratorch
613 # tifffile
726 - # torchgeo
727 - # torchmetrics
614 # torchvision
615 # transformers
616 # tritonclient
617 # vocos
732 - # xarray
618 # xgrammar
734 -omegaconf==2.3.0
735 - # via
736 - # hydra-core
737 - # lightning
619 open-clip-torch==2.32.0
620 # via -r requirements/test/rocm.in
621 openai==2.31.0
@@ -824,46 +705,29 @@ packaging==26.0
705 # datasets
706 # evaluate
707 # fastparquet
827 - # geopandas
708 # huggingface-hub
829 - # hydra-core
709 # kaleido
831 - # kornia
710 # lazy-loader
833 - # lightning
834 - # lightning-utilities
711 # lm-format-enforcer
712 # matplotlib
713 # optuna
714 # peft
715 # plotly
716 # pooch
841 - # pyogrio
717 # pytest
718 # pytest-rerunfailures
844 - # pytorch-lightning
719 # ray
846 - # rioxarray
720 # scikit-image
721 # statsmodels
849 - # tensorboard
850 - # tensorboardx
851 - # torchmetrics
722 # transformers
723 # typepy
854 - # wandb
855 - # xarray
724 pandas==3.0.1
725 # via
726 # datasets
727 # evaluate
728 # fastparquet
729 # genai-perf
862 - # geopandas
730 # statsmodels
864 - # tacoreader
865 - # torchgeo
866 - # xarray
731 partial-json-parser==0.2.1.1.post7
732 # via -r requirements/test/../common.txt
733 pathspec==1.0.4
@@ -881,18 +745,14 @@ perf-analyzer==0.1.0
745 pillow==12.1.1
746 # via
747 # -r requirements/test/../common.txt
884 - # diffusers
748 # genai-perf
749 # imagehash
750 # imageio
888 - # lightly-utils
751 # matplotlib
752 # mistral-common
753 # perceptron
754 # scikit-image
755 # segmentation-models-pytorch
894 - # tensorboard
895 - # torchgeo
756 # torchvision
757 platformdirs==4.3.6
758 # via
@@ -900,7 +760,6 @@ platformdirs==4.3.6
760 # pooch
761 # python-discovery
762 # virtualenv
903 - # wandb
763 plotly==6.6.0
764 # via
765 # -r requirements/test/rocm.in
@@ -946,10 +805,7 @@ protobuf==6.33.6
805 # opentelemetry-proto
806 # proto-plus
807 # ray
949 - # tensorboard
950 - # tensorboardx
808 # tensorizer
952 - # wandb
809 psutil==7.2.2
810 # via
811 # -r requirements/test/../common.txt
@@ -966,16 +822,12 @@ pyarrow==23.0.1
822 # via
823 # datasets
824 # genai-perf
969 - # tacoreader
970 - # terratorch
825 pyasn1==0.6.3
826 # via pyasn1-modules
827 pyasn1-modules==0.4.2
828 # via google-auth
829 pybase64==1.4.3
830 # via -r requirements/test/../common.txt
977 -pycocotools==2.0.11
978 - # via terratorch
831 pycountry==26.2.16
832 # via pydantic-extra-types
833 pycparser==3.0
@@ -994,7 +846,6 @@ pydantic==2.12.5
846 # fastapi
847 # fastapi-cloud-cli
848 # gpt-oss
997 - # lightly
849 # lm-format-enforcer
850 # mcp
851 # mistral-common
@@ -1005,7 +856,6 @@ pydantic==2.12.5
856 # pydantic-extra-types
857 # pydantic-settings
858 # ray
1008 - # wandb
859 # xgrammar
860 pydantic-core==2.41.5
861 # via pydantic
@@ -1023,17 +873,8 @@ pyjwt==2.12.1
873 # via
874 # mcp
875 # msal
1026 -pyogrio==0.12.1
1027 - # via geopandas
876 pyparsing==3.3.2
1029 - # via
1030 - # matplotlib
1031 - # rasterio
1032 -pyproj==3.7.2
1033 - # via
1034 - # geopandas
1035 - # rioxarray
1036 - # torchgeo
877 + # via matplotlib
878 pyrate-limiter==3.9.0
879 # via schemathesis
880 pystemmer==3.0.0
@@ -1070,13 +911,10 @@ pytest-subtests==0.14.2
911 # via schemathesis
912 pytest-timeout==2.3.1
913 # via -r requirements/test/rocm.in
1073 -python-box==7.4.1
1074 - # via terratorch
914 python-dateutil==2.9.0.post0
915 # via
916 # arrow
917 # botocore
1079 - # lightly
918 # matplotlib
919 # pandas
920 # typepy
@@ -1096,10 +934,6 @@ python-rapidjson==1.23
934 # via tritonclient
935 pytokens==0.4.1
936 # via black
1099 -pytorch-lightning==2.6.1
1100 - # via
1101 - # lightly
1102 - # lightning
937 pytrec-eval-terrier==0.5.10
938 # via mteb
939 pytz==2026.1.post1
@@ -1116,13 +950,9 @@ pyyaml==6.0.3
950 # genai-perf
951 # gguf
952 # huggingface-hub
1119 - # jsonargparse
1120 - # lightning
953 # lm-format-enforcer
1122 - # omegaconf
954 # optuna
955 # peft
1125 - # pytorch-lightning
956 # ray
957 # responses
958 # schemathesis
@@ -1130,7 +960,6 @@ pyyaml==6.0.3
960 # transformers
961 # uvicorn
962 # vocos
1133 - # wandb
963 pyzmq==27.1.0
964 # via
965 # -c requirements/common.txt
@@ -1139,11 +968,6 @@ rapidfuzz==3.12.1
968 # via
969 # -r requirements/test/rocm.in
970 # jiwer
1142 -rasterio==1.5.0
1143 - # via
1144 - # rioxarray
1145 - # terratorch
1146 - # torchgeo
971 ray==2.54.0
972 # via -r requirements/test/rocm.in
973 redis==7.3.0
@@ -1155,7 +979,6 @@ referencing==0.37.0
979 regex==2026.2.28
980 # via
981 # -r requirements/test/../common.txt
1158 - # diffusers
982 # nltk
983 # open-clip-torch
984 # sacrebleu
@@ -1168,14 +991,12 @@ requests==2.32.5
991 # azure-core
992 # buildkite-test-collector
993 # datasets
1171 - # diffusers
994 # docker
995 # evaluate
996 # gguf
997 # google-api-core
998 # google-cloud-storage
999 # gpt-oss
1178 - # lightly
1000 # lm-eval
1001 # mistral-common
1002 # msal
@@ -1186,9 +1007,7 @@ requests==2.32.5
1007 # responses
1008 # schemathesis
1009 # starlette-testclient
1189 - # tacoreader
1010 # tiktoken
1191 - # wandb
1011 responses==0.26.0
1012 # via genai-perf
1013 rfc3339-validator==0.1.4
@@ -1198,11 +1017,9 @@ rfc3987==1.3.8
1017 rich==14.3.3
1018 # via
1019 # genai-perf
1201 - # lightning
1020 # mteb
1021 # perceptron
1022 # rich-toolkit
1205 - # terratorch
1023 # typer
1024 rich-toolkit==0.19.7
1025 # via
@@ -1210,16 +1027,12 @@ rich-toolkit==0.19.7
1027 # fastapi-cloud-cli
1028 rignore==0.7.6
1029 # via fastapi-cloud-cli
1213 -rioxarray==0.22.0
1214 - # via terratorch
1030 rouge-score==0.1.2
1031 # via lm-eval
1032 rpds-py==0.30.0
1033 # via
1034 # jsonschema
1035 # referencing
1221 -rtree==1.4.1
1222 - # via torchgeo
1036 runai-model-streamer==0.15.7
1037 # via
1038 # -c requirements/rocm.txt
@@ -1237,7 +1050,6 @@ sacrebleu==2.6.0
1050 safetensors==0.7.0
1051 # via
1052 # accelerate
1240 - # diffusers
1053 # open-clip-torch
1054 # peft
1055 # segmentation-models-pytorch
@@ -1246,9 +1058,7 @@ safetensors==0.7.0
1058 schemathesis==3.39.15
1059 # via -r requirements/test/rocm.in
1060 scikit-image==0.26.0
1249 - # via
1250 - # albumentations
1251 - # terratorch
1061 + # via albumentations
1062 scikit-learn==1.8.0
1063 # via
1064 # albumentations
@@ -1256,7 +1066,6 @@ scikit-learn==1.8.0
1066 # lm-eval
1067 # mteb
1068 # sentence-transformers
1259 - # terratorch
1069 scipy==1.17.1
1070 # via
1071 # albumentations
@@ -1271,10 +1080,7 @@ scipy==1.17.1
1080 # statsmodels
1081 # vocos
1082 segmentation-models-pytorch==0.5.0
1274 - # via
1275 - # -r requirements/test/rocm.in
1276 - # terratorch
1277 - # torchgeo
1083 + # via -r requirements/test/rocm.in
1084 sentence-transformers==5.3.0
1085 # via
1086 # -r requirements/test/rocm.in
@@ -1282,9 +1088,7 @@ sentence-transformers==5.3.0
1088 sentencepiece==0.2.1
1089 # via -r requirements/test/../common.txt
1090 sentry-sdk==2.55.0
1285 - # via
1286 - # fastapi-cloud-cli
1287 - # wandb
1091 + # via fastapi-cloud-cli
1092 setproctitle==1.3.7
1093 # via -r requirements/test/../common.txt
1094 setuptools==79.0.1
@@ -1294,12 +1098,7 @@ setuptools==79.0.1
1098 # -r requirements/test/../common.txt
1099 # model-hosting-container-standards
1100 # pytablewriter
1297 - # tensorboard
1101 # torch
1299 -shapely==2.1.2
1300 - # via
1301 - # geopandas
1302 - # torchgeo
1102 shellingham==1.5.4
1103 # via
1104 # perceptron
@@ -1311,15 +1110,12 @@ six==1.17.0
1110 # -c requirements/common.txt
1111 # -r requirements/test/../common.txt
1112 # junit-xml
1314 - # lightly
1113 # opencensus
1114 # python-dateutil
1115 # rfc3339-validator
1116 # rouge-score
1117 smart-open==7.5.1
1118 # via ray
1321 -smmap==5.0.3
1322 - # via gitdb
1119 sniffio==1.3.1
1120 # via
1121 # anthropic
@@ -1358,8 +1154,6 @@ starlette-testclient==0.4.1
1154 # via schemathesis
1155 statsmodels==0.14.6
1156 # via genai-perf
1361 -stringzilla==4.6.0
1362 - # via albucore
1157 structlog==25.5.0
1158 # via gpt-oss
1159 supervisor==4.3.0
@@ -1372,8 +1166,6 @@ tabledata==1.3.4
1166 # via pytablewriter
1167 tabulate==0.10.0
1168 # via sacrebleu
1375 -tacoreader==0.5.6
1376 - # via terratorch
1169 tblib==3.1.0
1170 # via -r requirements/test/rocm.in
1171 tcolorpy==0.1.7
@@ -1382,28 +1174,16 @@ tenacity==9.1.4
1174 # via
1175 # gpt-oss
1176 # lm-eval
1385 -tensorboard==2.20.0
1386 - # via terratorch
1387 -tensorboard-data-server==0.7.2
1388 - # via tensorboard
1389 -tensorboardx==2.6.4
1390 - # via lightning
1177 tensorizer==2.10.1
1178 # via
1179 # -c requirements/rocm.txt
1180 # -r requirements/test/rocm.in
1181 termcolor==3.3.0
1396 - # via
1397 - # gpt-oss
1398 - # terratorch
1399 -terratorch==1.2.2
1400 - # via -r requirements/test/rocm.in
1182 + # via gpt-oss
1183 threadpoolctl==3.6.0
1184 # via scikit-learn
1185 tifffile==2026.3.3
1404 - # via
1405 - # scikit-image
1406 - # terratorch
1186 + # via scikit-image
1187 tiktoken==0.12.0
1188 # via
1189 # -c requirements/common.txt
@@ -1417,8 +1197,6 @@ timm==1.0.17
1197 # -r requirements/test/rocm.in
1198 # open-clip-torch
1199 # segmentation-models-pytorch
1420 - # terratorch
1421 - # torchgeo
1200 tokenizers==0.22.2
1201 # via
1202 # -c requirements/common.txt
@@ -1429,16 +1207,6 @@ tomli==2.4.0
1207 # via schemathesis
1208 tomli-w==1.2.0
1209 # via schemathesis
1432 -torchgeo==0.7.0
1433 - # via
1434 - # -r requirements/test/rocm.in
1435 - # terratorch
1436 -torchmetrics==1.9.0
1437 - # via
1438 - # lightning
1439 - # pytorch-lightning
1440 - # terratorch
1441 - # torchgeo
1210 tqdm==4.67.3
1211 # via
1212 # -r requirements/test/../common.txt
@@ -1446,8 +1214,6 @@ tqdm==4.67.3
1214 # evaluate
1215 # gguf
1216 # huggingface-hub
1449 - # lightly
1450 - # lightning
1217 # lm-eval
1218 # mteb
1219 # nltk
@@ -1456,11 +1222,8 @@ tqdm==4.67.3
1222 # optuna
1223 # peft
1224 # pqdm
1459 - # pytorch-lightning
1225 # segmentation-models-pytorch
1226 # sentence-transformers
1462 - # tacoreader
1463 - # terratorch
1227 # transformers
1228 transformers==5.5.3
1229 # via
@@ -1492,8 +1255,6 @@ typer==0.24.1
1255 # huggingface-hub
1256 # perceptron
1257 # transformers
1495 -typeshed-client==2.9.0
1496 - # via jsonargparse
1258 typing-extensions==4.15.0
1259 # via
1260 # -c requirements/common.txt
@@ -1511,8 +1272,6 @@ typing-extensions==4.15.0
1272 # grpcio
1273 # huggingface-hub
1274 # librosa
1514 - # lightning
1515 - # lightning-utilities
1275 # lm-eval
1276 # mcp
1277 # mistral-common
@@ -1527,18 +1286,14 @@ typing-extensions==4.15.0
1286 # pydantic
1287 # pydantic-core
1288 # pydantic-extra-types
1530 - # pytorch-lightning
1289 # referencing
1290 # rich-toolkit
1291 # sentence-transformers
1292 # sqlalchemy
1293 # starlette
1294 # torch
1537 - # torchgeo
1295 # typeguard
1539 - # typeshed-client
1296 # typing-inspection
1541 - # wandb
1297 # xgrammar
1298 typing-inspection==0.4.2
1299 # via
@@ -1555,7 +1310,6 @@ urllib3==2.6.3
1310 # blobfile
1311 # botocore
1312 # docker
1558 - # lightly
1313 # requests
1314 # responses
1315 # sentry-sdk
@@ -1575,8 +1329,6 @@ virtualenv==21.2.0
1329 # via ray
1330 vocos==0.1.0
1331 # via -r requirements/test/rocm.in
1578 -wandb==0.25.1
1579 - # via terratorch
1332 watchfiles==1.1.1
1333 # via
1334 # -r requirements/test/../common.txt
@@ -1588,15 +1340,11 @@ webcolors==25.10.0
1340 websockets==16.0
1341 # via uvicorn
1342 werkzeug==3.1.6
1591 - # via
1592 - # schemathesis
1593 - # tensorboard
1343 + # via schemathesis
1344 word2number==1.1
1345 # via lm-eval
1346 wrapt==2.1.2
1347 # via smart-open
1598 -xarray==2026.2.0
1599 - # via rioxarray
1348 xgrammar==0.1.33
1349 # via
1350 # -c requirements/common.txt
tests/entrypoints/openai/chat_completion/test_vision_embeds.py
+11
@@ -1,6 +1,8 @@
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4 +import importlib.util
5 +
6 import numpy as np
7 import pybase64 as base64
8 import pytest
@@ -10,7 +12,16 @@ import torch
12 from tests.utils import RemoteOpenAIServer
13 from vllm.utils.serial_utils import tensor2base64
14
15 +# Prithvi requires terratorch, which is temporarily unavailable while PyPI has
16 +# `lightning` quarantined (#41376). Skip just the Prithvi case; leave the
17 +# Qwen3-VL case in the same file untouched.
18 +_TERRATORCH_AVAILABLE = importlib.util.find_spec("terratorch") is not None
19 +
20
21 +@pytest.mark.skipif(
22 + not _TERRATORCH_AVAILABLE,
23 + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
24 +)
25 @pytest.mark.parametrize(
26 "model_name", ["ibm-nasa-geospatial/Prithvi-EO-2.0-300M-TL-Sen1Floods11"]
27 )
tests/models/multimodal/pooling/test_prithvi_mae.py
+5
@@ -6,6 +6,11 @@ import torch
6
7 from ....conftest import VllmRunner
8
9 +pytest.importorskip(
10 + "terratorch",
11 + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
12 +)
13 +
14
15 def _run_test(
16 vllm_runner: type[VllmRunner],
tests/models/test_terratorch.py
+5
@@ -7,6 +7,11 @@ import torch
7 from tests.conftest import VllmRunner
8 from tests.utils import create_new_process_for_each_test
9
10 +pytest.importorskip(
11 + "terratorch",
12 + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
13 +)
14 +
15
16 @create_new_process_for_each_test() # Hangs otherwise
17 @pytest.mark.parametrize(
tests/plugins_tests/test_terratorch_io_processor_plugins.py
+5
@@ -11,6 +11,11 @@ from PIL import Image
11 from tests.utils import RemoteOpenAIServer
12 from vllm.entrypoints.pooling.pooling.protocol import IOProcessorResponse
13
14 +pytest.importorskip(
15 + "terratorch",
16 + reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
17 +)
18 +
19 models_config = {
20 "ibm-nasa-geospatial/Prithvi-EO-2.0-300M-TL-Sen1Floods11": {
21 "image_url": "https://huggingface.co/christian-pinto/Prithvi-EO-2.0-300M-TL-VLLM/resolve/main/valencia_example_2024-10-26.tiff", # noqa: E501