fix: Allow non-lro delete method

PiperOrigin-RevId: 640765520
googleapis · Jun 6, 2024 · c23c0ad · c23c0ad
1 parent 70168eb
commit c23c0ad
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/google/cloud/aiplatform/base.py b/google/cloud/aiplatform/base.py
@@ -1199,12 +1199,16 @@ def _list_with_local_order(
     def _delete(self) -> None:
         """Deletes this Vertex AI resource. WARNING: This deletion is permanent."""
         _LOGGER.log_action_start_against_resource("Deleting", "", self)
-        lro = getattr(self.api_client, self._delete_method)(name=self.resource_name)
-        _LOGGER.log_action_started_against_resource_with_lro(
-            "Delete", "", self.__class__, lro
+        possible_lro = getattr(self.api_client, self._delete_method)(
+            name=self.resource_name
         )
-        lro.result()
-        _LOGGER.log_action_completed_against_resource("deleted.", "", self)
+
+        if possible_lro:
+            _LOGGER.log_action_started_against_resource_with_lro(
+                "Delete", "", self.__class__, possible_lro
+            )
+            possible_lro.result()
+            _LOGGER.log_action_completed_against_resource("deleted.", "", self)
 
 
 class VertexAiResourceNounWithFutureManager(_VertexAiResourceNounPlus, FutureManager):