Consolidate more _WaitUntilReady logic

PiperOrigin-RevId: 829634862

Author: hubatish

perfkitbenchmarker/providers/gcp/vertex_ai.py

@@ -22,7 +22,6 @@
 from perfkitbenchmarker import resource
 from perfkitbenchmarker import sample
 from perfkitbenchmarker import virtual_machine
-from perfkitbenchmarker import vm_util
 from perfkitbenchmarker.providers.gcp import flags as gcp_flags
 from perfkitbenchmarker.providers.gcp import gcs
 from perfkitbenchmarker.providers.gcp import util
@@ -419,18 +418,7 @@ def _DeployModel(self):
           'The operations may still be underway remotely and may still succeed'
           in err
       ):
-
-        @vm_util.Retry(
-            poll_interval=self.POLL_INTERVAL,
-            fuzz=0,
-            timeout=self.READY_TIMEOUT,
-            retryable_exceptions=(errors.Resource.RetryableCreationError,),
-        )
-        def WaitUntilReady():
-          if not self._IsReady():
-            raise errors.Resource.RetryableCreationError('Not yet ready')
-
-        WaitUntilReady()
+        self._WaitUntilReady()
       elif 'Machine type temporarily unavailable' in err:
         raise errors.Benchmarks.QuotaFailure(err)
       else: