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 Prithvi/Terratorch model-registry tests when terratorch is missing (#41389)
Signed-off-by: Stefano Castagnetta <scastagnetta@nvidia.com>
Stefano Castagnetta committed
Apr 30, 2026 at 21:47 UTC
efb4cdf2b8000c850d04706eb6f788903e3ee544
5 files changed
+32
-6
tests/models/multimodal/pooling/test_prithvi_mae.py
+4
-2
@@ -1,13 +1,15 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4
+import importlib.util
5
+
6
import pytest
7
import torch
8
9
from ....conftest import VllmRunner
10
9
-pytest.importorskip(
10
- "terratorch",
11
+pytestmark = pytest.mark.skipif(
12
+ importlib.util.find_spec("terratorch") is None,
13
reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
14
)
15
tests/models/test_initialization.py
+10
@@ -109,6 +109,16 @@ def can_initialize(
109
"which is not configured in test environment"
110
)
111
112
+ if model_arch in ("PrithviGeoSpatialMAE", "Terratorch"):
113
+ import importlib.util
114
+
115
+ if importlib.util.find_spec("terratorch") is None:
116
+ pytest.skip(
117
+ "terratorch is not installed; "
118
+ "temporarily skipped while PyPI has `lightning` quarantined "
119
+ "(see #41376)"
120
+ )
121
+
122
if model_arch in ["DeepseekV32ForCausalLM", "GlmMoeDsaForCausalLM"]:
123
from vllm.platforms import current_platform
124
tests/models/test_registry.py
+11
@@ -36,6 +36,17 @@ def test_registry_imports(model_arch):
36
check_max_version=False,
37
check_version_reason="vllm",
38
)
39
+
40
+ if model_arch in ("PrithviGeoSpatialMAE", "Terratorch"):
41
+ import importlib.util
42
+
43
+ if importlib.util.find_spec("terratorch") is None:
44
+ pytest.skip(
45
+ "terratorch is not installed; "
46
+ "temporarily skipped while PyPI has `lightning` quarantined "
47
+ "(see #41376)"
48
+ )
49
+
50
# Ensure all model classes can be imported successfully
51
model_cls = ModelRegistry._try_load_model_cls(model_arch)
52
assert model_cls is not None
tests/models/test_terratorch.py
+4
-2
@@ -1,14 +1,16 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4
+import importlib.util
5
+
6
import pytest
7
import torch
8
9
from tests.conftest import VllmRunner
10
from tests.utils import create_new_process_for_each_test
11
10
-pytest.importorskip(
11
- "terratorch",
12
+pytestmark = pytest.mark.skipif(
13
+ importlib.util.find_spec("terratorch") is None,
14
reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
15
)
16
tests/plugins_tests/test_terratorch_io_processor_plugins.py
+3
-2
@@ -1,5 +1,6 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
+import importlib.util
4
import io
5
6
import imagehash
@@ -11,8 +12,8 @@ from PIL import Image
12
from tests.utils import RemoteOpenAIServer
13
from vllm.entrypoints.pooling.pooling.protocol import IOProcessorResponse
14
14
-pytest.importorskip(
15
- "terratorch",
15
+pytestmark = pytest.mark.skipif(
16
+ importlib.util.find_spec("terratorch") is None,
17
reason="terratorch unavailable while PyPI has `lightning` quarantined; see #41376",
18
)
19