From f08c19b1eb94e60eedb1dd5bd0f4a1c148e871bf Mon Sep 17 00:00:00 2001
From: Alexey Volkov <alexey.volkov@ark-kun.com>
Date: Fri, 4 Feb 2022 17:28:50 -0800
Subject: [PATCH] Fixed getting the class name.

---
 google/cloud/aiplatform/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/google/cloud/aiplatform/base.py b/google/cloud/aiplatform/base.py
index b45331a958..dcec7bb5d6 100644
--- a/google/cloud/aiplatform/base.py
+++ b/google/cloud/aiplatform/base.py
@@ -677,7 +677,7 @@ def to_dict(self) -> Dict[str, Any]:
     def _generate_display_name(cls, prefix: Optional[str] = None) -> str:
         """Returns a display name containing class name and time string."""
         if not prefix:
-            prefix = cls.name
+            prefix = cls.__name__
         return prefix + " " + datetime.datetime.now().isoformat(sep=" ")