Skip to content

Commit a9048db

Browse files
authored
test: Improve stability of server shutdown test in L0_python_api::test_api::test_stop (#383)
1 parent d2abb8b commit a9048db

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

python/test/test_api.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@
7070
exit_on_error=True,
7171
strict_model_config=False,
7272
model_control_mode=tritonserver.ModelControlMode.EXPLICIT,
73-
exit_timeout=10,
73+
exit_timeout=30,
7474
)
7575

7676

@@ -357,6 +357,11 @@ def test_stop(self):
357357
{
358358
"backend": "python",
359359
"parameters": {"decoupled": {"string_value": "False"}},
360+
# Keep instance count low for fast startup/cleanup.
361+
# Alternatively can use KIND_CPU here, but keeping gpus/count explicit.
362+
"instance_group": [
363+
{"kind": "KIND_GPU", "gpus": [0], "count": 1}
364+
],
360365
}
361366
)
362367
},

0 commit comments

Comments
 (0)