Fixed validation

googleapis · Jan 13, 2022 · 4a4a6b6 · 4a4a6b6
1 parent b2ae762
commit 4a4a6b6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -1539,13 +1539,12 @@ def update(
             ValueError: If `labels` is not the correct format.
         """
 
-        if labels:
-            utils.validate_labels(labels)
-
         current_model_proto = self.gca_resource
         update_mask: List[str] = []
 
         if display_name:
+            utils.validate_display_name(display_name)
+
             current_model_proto.display_name = display_name
             update_mask.append("display_name")