Skip to content

Commit fb99d80

Browse files
DevakiBolleneniDevakiBolleneni
andauthored
Fix hf-vllm mainline test logic (#5635)
* fix hf-vllm mainline test logic * fix formatting --------- Co-authored-by: DevakiBolleneni <devakib@amazon.com>
1 parent 48bf00c commit fb99d80

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

test/testrunner.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,9 @@ def main():
282282
f"NOTE: {specific_test_type} tests not supported on base images. Skipping..."
283283
)
284284
return
285-
elif all("vllm" in image_uri for image_uri in all_image_list) and test_type not in {
285+
elif all(
286+
"vllm" in image_uri and "huggingface" not in image_uri for image_uri in all_image_list
287+
) and test_type not in {
286288
"functionality_sanity",
287289
"security_sanity",
288290
"eks",
@@ -293,6 +295,18 @@ def main():
293295
f"NOTE: {specific_test_type} tests not supported on vllm images. Skipping..."
294296
)
295297
return
298+
elif all(
299+
"huggingface" in image_uri and "vllm" in image_uri for image_uri in all_image_list
300+
) and test_type not in {
301+
"functionality_sanity",
302+
"security_sanity",
303+
"sagemaker",
304+
"sagemaker-local",
305+
}:
306+
LOGGER.info(
307+
f"NOTE: {specific_test_type} tests not supported on huggingface-vllm images. Skipping..."
308+
)
309+
return
296310
elif all("sglang" in image_uri for image_uri in all_image_list) and test_type not in {
297311
"functionality_sanity",
298312
"security_sanity",

0 commit comments

Comments
 (0)