From ea043eb9e185e1ba8f7c8badf888686e7c4ee81f Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Wed, 24 Jul 2024 15:12:06 -0700 Subject: [PATCH] Extend exit timeout to standard 30 seconds, and keep instance count explicitly to 1 to avoid long shutdown times on CI machines with many GPUs --- python/test/test_api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/test/test_api.py b/python/test/test_api.py index 93124c470..452429ed9 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,8 @@ def test_stop(self): { "backend": "python", "parameters": {"decoupled": {"string_value": "False"}}, + # Keep instance count low for fast startup/cleanup + "instance_group": [{"kind": "KIND_GPU", "count": 1}], } ) },