[CI] re-enable test_vision_openai_server_a ci (#13444)
This commit is contained in:
@@ -124,7 +124,7 @@ suites = {
|
|||||||
TestFile("test_triton_sliding_window.py", 100),
|
TestFile("test_triton_sliding_window.py", 100),
|
||||||
TestFile("test_utils_update_weights.py", 48),
|
TestFile("test_utils_update_weights.py", 48),
|
||||||
TestFile("test_vision_chunked_prefill.py", 170),
|
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_vlm_input_format.py", 300),
|
||||||
TestFile("test_modelopt_loader.py", 30),
|
TestFile("test_modelopt_loader.py", 30),
|
||||||
TestFile("test_modelopt_export.py", 30),
|
TestFile("test_modelopt_export.py", 30),
|
||||||
@@ -365,7 +365,7 @@ suites = {
|
|||||||
TestFile("test_verl_engine_4_gpu.py"),
|
TestFile("test_verl_engine_4_gpu.py"),
|
||||||
TestFile("test_verl_engine_server.py"),
|
TestFile("test_verl_engine_server.py"),
|
||||||
TestFile("test_vertex_endpoint.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_b.py"),
|
||||||
TestFile("test_vision_openai_server_common.py"),
|
TestFile("test_vision_openai_server_common.py"),
|
||||||
TestFile("test_vlm_accuracy.py"),
|
TestFile("test_vlm_accuracy.py"),
|
||||||
|
|||||||
@@ -6,7 +6,16 @@ python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_multi_
|
|||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
|
import openai
|
||||||
from test_vision_openai_server_common import *
|
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):
|
class TestLlavaServer(ImageOpenAITestMixin):
|
||||||
@@ -207,12 +216,11 @@ class TestDeepseekOCRServer(TestOpenAIMLLMServerBase):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# Note: Cannot delete mixin classes imported via * since they're not in local scope
|
del (
|
||||||
# del (
|
TestOpenAIMLLMServerBase,
|
||||||
# TestOpenAIMLLMServerBase,
|
ImageOpenAITestMixin,
|
||||||
# ImageOpenAITestMixin,
|
VideoOpenAITestMixin,
|
||||||
# VideoOpenAITestMixin,
|
AudioOpenAITestMixin,
|
||||||
# AudioOpenAITestMixin,
|
OmniOpenAITestMixin,
|
||||||
# OmniOpenAITestMixin,
|
)
|
||||||
# )
|
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|||||||
Reference in New Issue
Block a user