diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 504515273..a83a0df20 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -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"), diff --git a/test/srt/test_vision_openai_server_a.py b/test/srt/test_vision_openai_server_a.py index 7f8cc87b5..5068dcda1 100644 --- a/test/srt/test_vision_openai_server_a.py +++ b/test/srt/test_vision_openai_server_a.py @@ -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()