diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 664dba7828c7..7c98bd401745 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -126,7 +126,7 @@ 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 @@ 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"), diff --git a/test/srt/test_vision_openai_server_a.py b/test/srt/test_vision_openai_server_a.py index 7f8cc87b5c19..5068dcda1a09 100644 --- a/test/srt/test_vision_openai_server_a.py +++ b/test/srt/test_vision_openai_server_a.py @@ -6,7 +6,16 @@ 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 @@ def test_single_image_chat_completion(self): 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()