From a3da19aac6bdd3fa8d218408582205f7241a4b04 Mon Sep 17 00:00:00 2001 From: Sam Goodman Date: Fri, 27 May 2022 13:17:40 -0700 Subject: [PATCH] fix: Changed system test to use list_models() correctly (#1397) * chore: release 1.13.1 Release-As: 1.13.1 * Changed test to use list_models() correctly * feat: Changed system test to use list_models() correctly Co-authored-by: Sam Goodman --- tests/system/aiplatform/test_model_upload.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/system/aiplatform/test_model_upload.py b/tests/system/aiplatform/test_model_upload.py index 3187453561..48e6169af4 100644 --- a/tests/system/aiplatform/test_model_upload.py +++ b/tests/system/aiplatform/test_model_upload.py @@ -77,9 +77,9 @@ def test_upload_and_deploy_xgboost_model(self, shared_state): assert model.description == "new_description" assert model.labels == {"my_label": "updated"} - assert len(endpoint.list_models) == 1 + assert len(endpoint.list_models()) == 1 endpoint.deploy(model, traffic_percentage=100) - assert len(endpoint.list_models) == 2 + assert len(endpoint.list_models()) == 2 traffic_split = { deployed_model.id: 50 for deployed_model in endpoint.list_models() }