Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
zhaoyu6
sglang
Commits
51e87f6f
"vscode:/vscode.git/clone" did not exist on "2c95db34d9e3ad76446b4537f401c28fe37c5e37"
Unverified
Commit
51e87f6f
authored
Jan 20, 2025
by
Lianmin Zheng
Committed by
GitHub
Jan 20, 2025
Browse files
Skip flaky custom_logit_processor tests (#3004)
parent
09bcbe01
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
test/srt/test_srt_endpoint.py
test/srt/test_srt_endpoint.py
+4
-0
No files found.
test/srt/test_srt_endpoint.py
View file @
51e87f6f
...
@@ -301,10 +301,14 @@ class TestSRTEndpoint(unittest.TestCase):
...
@@ -301,10 +301,14 @@ class TestSRTEndpoint(unittest.TestCase):
def
test_custom_logit_processor
(
self
):
def
test_custom_logit_processor
(
self
):
"""Test custom logit processor with a single request."""
"""Test custom logit processor with a single request."""
# Temporarily skipped due to buggy implementation
return
self
.
run_custom_logit_processor
(
target_token_id
=
5
)
self
.
run_custom_logit_processor
(
target_token_id
=
5
)
def
test_custom_logit_processor_batch
(
self
):
def
test_custom_logit_processor_batch
(
self
):
"""Test custom logit processor with a batch of requests."""
"""Test custom logit processor with a batch of requests."""
# Temporarily skipped due to buggy implementation
return
target_token_ids
=
list
(
range
(
32
))
target_token_ids
=
list
(
range
(
32
))
with
ThreadPoolExecutor
(
len
(
target_token_ids
))
as
executor
:
with
ThreadPoolExecutor
(
len
(
target_token_ids
))
as
executor
:
list
(
executor
.
map
(
self
.
run_custom_logit_processor
,
target_token_ids
))
list
(
executor
.
map
(
self
.
run_custom_logit_processor
,
target_token_ids
))
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment