From a9048db96d270ec0adb75e27b00997f8e9e383cd Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Thu, 25 Jul 2024 12:28:44 -0700 Subject: [PATCH] test: Improve stability of server shutdown test in L0_python_api::test_api::test_stop (#383) --- python/test/test_api.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/test/test_api.py b/python/test/test_api.py index 93124c470..c15847aab 100644 --- a/python/test/test_api.py +++ b/python/test/test_api.py @@ -70,7 +70,7 @@ exit_on_error=True, strict_model_config=False, model_control_mode=tritonserver.ModelControlMode.EXPLICIT, - exit_timeout=10, + exit_timeout=30, ) @@ -357,6 +357,11 @@ def test_stop(self): { "backend": "python", "parameters": {"decoupled": {"string_value": "False"}}, + # Keep instance count low for fast startup/cleanup. + # Alternatively can use KIND_CPU here, but keeping gpus/count explicit. + "instance_group": [ + {"kind": "KIND_GPU", "gpus": [0], "count": 1} + ], } ) },