diff --git a/qa/python_models/bls_model_loading/config.pbtxt b/qa/python_models/bls_model_loading/config.pbtxt index 282746a4d9..2099ba5db7 100644 --- a/qa/python_models/bls_model_loading/config.pbtxt +++ b/qa/python_models/bls_model_loading/config.pbtxt @@ -37,7 +37,7 @@ output [ instance_group [ { - count: 3 + count: 1 kind: KIND_CPU } ] diff --git a/qa/python_models/bls_model_loading/model.py b/qa/python_models/bls_model_loading/model.py index ea9d3c1c94..4c4339ce5c 100644 --- a/qa/python_models/bls_model_loading/model.py +++ b/qa/python_models/bls_model_loading/model.py @@ -35,6 +35,8 @@ def setUp(self): self.model_name = "onnx_int32_int32_int32" def tearDown(self): + # The unload call does not wait for the requested model to be fully + # unloaded before returning. pb_utils.unload_model(self.model_name) def test_load_unload_model(self):