From 17b9e8fa4e469479c09e6bbfd4e2007b6fdbb9a3 Mon Sep 17 00:00:00 2001 From: Morgan Du Date: Wed, 6 Apr 2022 15:57:49 -0700 Subject: [PATCH] fix: system tests failure due to model upload --- google/cloud/aiplatform/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py index ab22d20f10..c934a01eec 100644 --- a/google/cloud/aiplatform/models.py +++ b/google/cloud/aiplatform/models.py @@ -2775,7 +2775,7 @@ def upload_xgboost_model_file( Also if model directory does not contain a supported model file. """ if not display_name: - display_name = cls.__class__.__generate_display_name("XGBoost model") + display_name = cls._generate_display_name("XGBoost model") XGBOOST_SUPPORTED_MODEL_FILE_EXTENSIONS = [ ".pkl",