Fixed getting the class name.

googleapis · Feb 5, 2022 · f08c19b · f08c19b
1 parent c0d2aee
commit f08c19b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=" ")