Skip to content

Commit

Permalink
Extend exit timeout to standard 30 seconds, and keep instance count e…
Browse files Browse the repository at this point in the history
…xplicitly to 1 to avoid long shutdown times on CI machines with many GPUs
  • Loading branch information
rmccorm4 committed Jul 24, 2024
1 parent d2abb8b commit ea043eb
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/test/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@
exit_on_error=True,
strict_model_config=False,
model_control_mode=tritonserver.ModelControlMode.EXPLICIT,
exit_timeout=10,
exit_timeout=30,
)


Expand Down Expand Up @@ -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}],
}
)
},
Expand Down

0 comments on commit ea043eb

Please sign in to comment.