fix: Unbreak additional timeout for MatchingEngine update_embeddings

matthew29tang · copybara-github · commit 5d0bc1ea24cd · 2023-01-25T10:54:44.000-08:00
PiperOrigin-RevId: 504606042
diff --git a/google/cloud/aiplatform/matching_engine/matching_engine_index.py b/google/cloud/aiplatform/matching_engine/matching_engine_index.py
@@ -347,7 +347,7 @@ def update_embeddings(
             "Update", "index", self.__class__, update_lro
         )
 
-        self._gca_resource = update_lro.result()
+        self._gca_resource = update_lro.result(timeout=None)
 
         _LOGGER.log_action_completed_against_resource("index", "Updated", self)
 

Original file line number	Diff line number	Diff line change
`@@ -347,7 +347,7 @@ def update_embeddings(`
`347`	`347`	`"Update", "index", self.__class__, update_lro`
`348`	`348`	`)`
`349`	`349`
`350`		`- self._gca_resource = update_lro.result()`
	`350`	`+ self._gca_resource = update_lro.result(timeout=None)`
`351`	`351`
`352`	`352`	`_LOGGER.log_action_completed_against_resource("index", "Updated", self)`
`353`	`353`