Skip to content

Commit a54a4a2

Browse files
yhyang20100INDEX
authored andcommitted
[CI] re-enable test_vision_openai_server_a ci (sgl-project#13444)
1 parent 98e347d commit a54a4a2

File tree

2 files changed

+18
-10
lines changed

2 files changed

+18
-10
lines changed

test/srt/run_suite.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@
124124
TestFile("test_triton_sliding_window.py", 100),
125125
TestFile("test_utils_update_weights.py", 48),
126126
TestFile("test_vision_chunked_prefill.py", 170),
127-
# TestFile("test_vision_openai_server_a.py", 900),
127+
TestFile("test_vision_openai_server_a.py", 900),
128128
TestFile("test_vlm_input_format.py", 300),
129129
TestFile("test_modelopt_loader.py", 30),
130130
TestFile("test_modelopt_export.py", 30),
@@ -365,7 +365,7 @@
365365
TestFile("test_verl_engine_4_gpu.py"),
366366
TestFile("test_verl_engine_server.py"),
367367
TestFile("test_vertex_endpoint.py"),
368-
TestFile("test_vision_openai_server_a.py"), # TODO: Fix timeout
368+
# TestFile("test_vision_openai_server_a.py"), # TODO: Fix timeout
369369
TestFile("test_vision_openai_server_b.py"),
370370
TestFile("test_vision_openai_server_common.py"),
371371
TestFile("test_vlm_accuracy.py"),

test/srt/test_vision_openai_server_a.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,16 @@
66

77
import unittest
88

9+
import openai
910
from test_vision_openai_server_common import *
11+
from test_vision_openai_server_common import ( # DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,; DEFAULT_URL_FOR_TEST,; IMAGE_MAN_IRONING_URL,; popen_launch_server,
12+
AudioOpenAITestMixin,
13+
CustomTestCase,
14+
ImageOpenAITestMixin,
15+
OmniOpenAITestMixin,
16+
TestOpenAIMLLMServerBase,
17+
VideoOpenAITestMixin,
18+
)
1019

1120

1221
class TestLlavaServer(ImageOpenAITestMixin):
@@ -207,12 +216,11 @@ def test_single_image_chat_completion(self):
207216

208217

209218
if __name__ == "__main__":
210-
# Note: Cannot delete mixin classes imported via * since they're not in local scope
211-
# del (
212-
# TestOpenAIMLLMServerBase,
213-
# ImageOpenAITestMixin,
214-
# VideoOpenAITestMixin,
215-
# AudioOpenAITestMixin,
216-
# OmniOpenAITestMixin,
217-
# )
219+
del (
220+
TestOpenAIMLLMServerBase,
221+
ImageOpenAITestMixin,
222+
VideoOpenAITestMixin,
223+
AudioOpenAITestMixin,
224+
OmniOpenAITestMixin,
225+
)
218226
unittest.main()

0 commit comments

Comments
 (0)