Skip to content

Commit fcc0a31

Browse files
authored
[CI] Fix tensorizer test assertion (vllm-project#24545)
Signed-off-by: Peter Schuurman <[email protected]>
1 parent 736569d commit fcc0a31

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/tensorizer_loader/test_tensorizer.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,11 +161,11 @@ def test_load_without_tensorizer_load_format(vllm_runner, capfd, model_ref):
161161
model = vllm_runner(
162162
model_ref,
163163
model_loader_extra_config=TensorizerConfig(tensorizer_uri="test"))
164+
pytest.fail("Expected RuntimeError for extra config keys")
164165
except RuntimeError:
165166
out, err = capfd.readouterr()
166167
combined_output = out + err
167-
assert ("ValueError: Model loader extra config "
168-
"is not supported for load "
168+
assert ("ValueError: Unexpected extra config keys for load "
169169
"format auto") in combined_output
170170
finally:
171171
del model
@@ -181,11 +181,12 @@ def test_raise_value_error_on_invalid_load_format(vllm_runner, capfd,
181181
model_ref,
182182
load_format="safetensors",
183183
model_loader_extra_config=TensorizerConfig(tensorizer_uri="test"))
184+
pytest.fail("Expected RuntimeError for extra config keys")
184185
except RuntimeError:
185186
out, err = capfd.readouterr()
186187

187188
combined_output = out + err
188-
assert ("ValueError: Model loader extra config is not supported "
189+
assert ("ValueError: Unexpected extra config keys "
189190
"for load format safetensors") in combined_output
190191
finally:
191192
del model

0 commit comments

Comments
 (0)