[CI] re-enable test_vision_openai_server_a ci (#13444)

This commit is contained in:
Yuhao Yang
2025-11-17 18:32:59 -08:00
committed by GitHub
parent fe3bbfb40c
commit 7119d188f6
2 changed files with 18 additions and 10 deletions
+2 -2
View File
@@ -124,7 +124,7 @@ suites = {
TestFile("test_triton_sliding_window.py", 100),
TestFile("test_utils_update_weights.py", 48),
TestFile("test_vision_chunked_prefill.py", 170),
# TestFile("test_vision_openai_server_a.py", 900),
TestFile("test_vision_openai_server_a.py", 900),
TestFile("test_vlm_input_format.py", 300),
TestFile("test_modelopt_loader.py", 30),
TestFile("test_modelopt_export.py", 30),
@@ -365,7 +365,7 @@ suites = {
TestFile("test_verl_engine_4_gpu.py"),
TestFile("test_verl_engine_server.py"),
TestFile("test_vertex_endpoint.py"),
TestFile("test_vision_openai_server_a.py"), # TODO: Fix timeout
# TestFile("test_vision_openai_server_a.py"), # TODO: Fix timeout
TestFile("test_vision_openai_server_b.py"),
TestFile("test_vision_openai_server_common.py"),
TestFile("test_vlm_accuracy.py"),
+16 -8
View File
@@ -6,7 +6,16 @@ python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_multi_
import unittest
import openai
from test_vision_openai_server_common import *
from test_vision_openai_server_common import ( # DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,; DEFAULT_URL_FOR_TEST,; IMAGE_MAN_IRONING_URL,; popen_launch_server,
AudioOpenAITestMixin,
CustomTestCase,
ImageOpenAITestMixin,
OmniOpenAITestMixin,
TestOpenAIMLLMServerBase,
VideoOpenAITestMixin,
)
class TestLlavaServer(ImageOpenAITestMixin):
@@ -207,12 +216,11 @@ class TestDeepseekOCRServer(TestOpenAIMLLMServerBase):
if __name__ == "__main__":
# Note: Cannot delete mixin classes imported via * since they're not in local scope
# del (
# TestOpenAIMLLMServerBase,
# ImageOpenAITestMixin,
# VideoOpenAITestMixin,
# AudioOpenAITestMixin,
# OmniOpenAITestMixin,
# )
del (
TestOpenAIMLLMServerBase,
ImageOpenAITestMixin,
VideoOpenAITestMixin,
AudioOpenAITestMixin,
OmniOpenAITestMixin,
)
unittest.main()