fix: system tests failure from test_upload_and_deploy_xgboost_model (#…

…1149) Fixes #1138 🦕
googleapis · Apr 7, 2022 · c8422a9 · c8422a9
1 parent 31b4f86
commit c8422a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",