[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_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"),
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user