From 8a28d7ae459f36bb72eba27cf14f6b3ef0f9abf5 Mon Sep 17 00:00:00 2001
From: Abhishek Vilas Munagekar <10258799+munagekar@users.noreply.github.com>
Date: Wed, 18 May 2022 08:34:55 +0900
Subject: [PATCH] fix tests: wait for LRO to complete even if not sync

---
 tests/unit/aiplatform/test_endpoints.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/unit/aiplatform/test_endpoints.py b/tests/unit/aiplatform/test_endpoints.py
index 8344db5b72..826be28c41 100644
--- a/tests/unit/aiplatform/test_endpoints.py
+++ b/tests/unit/aiplatform/test_endpoints.py
@@ -1182,6 +1182,9 @@ def test_deploy_with_autoscaling_target_accelerator_duty_cycle_and_no_accelerato
                 autoscaling_target_accelerator_duty_cycle=70,
             )
 
+            if not sync:
+                test_endpoint.wait()
+
     @pytest.mark.usefixtures("get_endpoint_mock", "get_model_mock")
     @pytest.mark.parametrize("sync", [True, False])
     def test_deploy_with_explanations(self, deploy_model_with_explanations_mock, sync):