fix tests: wait for LRO to complete even if not sync

googleapis · May 17, 2022 · 8a28d7a · 8a28d7a
1 parent 0f2353e
commit 8a28d7a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):