Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
dynamo
Commits
002617a9
"launch/vscode:/vscode.git/clone" did not exist on "c443528fbac586ceecdba0a2841ba82841340dba"
Unverified
Commit
002617a9
authored
Sep 17, 2025
by
Richard Huo
Committed by
GitHub
Sep 17, 2025
Browse files
fix: fix some trtllm tests (#3084)
parent
eb6722e3
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
2 deletions
+5
-2
tests/serve/test_trtllm.py
tests/serve/test_trtllm.py
+1
-2
tests/utils/engine_process.py
tests/utils/engine_process.py
+4
-0
No files found.
tests/serve/test_trtllm.py
View file @
002617a9
...
@@ -59,7 +59,6 @@ trtllm_configs = {
...
@@ -59,7 +59,6 @@ trtllm_configs = {
request_payloads
=
[
request_payloads
=
[
chat_payload_default
(
chat_payload_default
(
expected_log
=
[
expected_log
=
[
r
"ZMQ listener .* received batch with \d+ events \(seq=\d+\)"
,
r
"Event processor for worker_id \d+ processing event: Stored\("
,
r
"Event processor for worker_id \d+ processing event: Stored\("
,
r
"Selected worker: \d+, logit: "
,
r
"Selected worker: \d+, logit: "
,
]
]
...
@@ -123,7 +122,7 @@ def test_chat_only_aggregated_with_test_logits_processor(
...
@@ -123,7 +122,7 @@ def test_chat_only_aggregated_with_test_logits_processor(
script_name
=
base
.
script_name
,
# agg.sh
script_name
=
base
.
script_name
,
# agg.sh
marks
=
[],
# not used by this direct test
marks
=
[],
# not used by this direct test
request_payloads
=
[
request_payloads
=
[
chat_payload_default
(),
chat_payload_default
(
expected_response
=
[
"Hello world!"
]
),
],
],
model
=
"Qwen/Qwen3-0.6B"
,
model
=
"Qwen/Qwen3-0.6B"
,
delayed_start
=
base
.
delayed_start
,
delayed_start
=
base
.
delayed_start
,
...
...
tests/utils/engine_process.py
View file @
002617a9
...
@@ -4,6 +4,7 @@
...
@@ -4,6 +4,7 @@
import
json
import
json
import
logging
import
logging
import
os
import
os
import
time
from
dataclasses
import
dataclass
,
field
from
dataclasses
import
dataclass
,
field
from
typing
import
Any
,
Dict
,
List
,
Optional
from
typing
import
Any
,
Dict
,
List
,
Optional
...
@@ -110,6 +111,9 @@ class EngineProcess(ManagedProcess):
...
@@ -110,6 +111,9 @@ class EngineProcess(ManagedProcess):
# Optionally validate expected log patterns after response handling
# Optionally validate expected log patterns after response handling
if
payload
.
expected_log
:
if
payload
.
expected_log
:
time
.
sleep
(
0.5
)
# The kv event sometimes needs extra time to arrive and be reflected in the log.
self
.
validate_expected_logs
(
payload
.
expected_log
)
self
.
validate_expected_logs
(
payload
.
expected_log
)
def
validate_expected_logs
(
self
,
patterns
:
Any
)
->
None
:
def
validate_expected_logs
(
self
,
patterns
:
Any
)
->
None
:
...
...
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