File tree Expand file tree Collapse file tree 1 file changed +15
-1
lines changed
Expand file tree Collapse file tree 1 file changed +15
-1
lines changed Original file line number Diff line number Diff 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" ,
You can’t perform that action at this time.
0 commit comments