diff --git a/tests/system/aiplatform/test_model_upload.py b/tests/system/aiplatform/test_model_upload.py index 625a3dc2c4..70f78aec60 100644 --- a/tests/system/aiplatform/test_model_upload.py +++ b/tests/system/aiplatform/test_model_upload.py @@ -72,5 +72,5 @@ def test_upload_and_deploy_xgboost_model(self, shared_state): labels={"my_label": "updated"}, ) assert model.display_name == "new_name" - assert model.display_name == "new_description" + assert model.description == "new_description" assert model.labels == {"my_label": "updated"}