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 <[email protected]>
googleapis · May 27, 2022 · a3da19a · a3da19a
1 parent 8560fa8
commit a3da19a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()
         }