fix: Changed system test to use list_models() correctly (#1397)

samgoodman · Sam Goodman · web-flow · commit a3da19aac6bd · 2022-05-27T13:17:40.000-07:00
* 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 &lt;goodmansam@google.com&gt;
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()
         }
