[vision] fix: add timeout for LRO result and mark it as flaky [(#3684)](

GoogleCloudPlatform/python-docs-samples#3684) fixes #3674
busunkim96 · May 4, 2020 · 974370a · 974370a
1 parent fbe33c4
commit 974370a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/samples/snippets/detect/vision_async_batch_annotate_images.py b/samples/snippets/detect/vision_async_batch_annotate_images.py
@@ -46,7 +46,7 @@ def sample_async_batch_annotate_images(
     operation = client.async_batch_annotate_images(requests, output_config)
 
     print("Waiting for operation to complete...")
-    response = operation.result()
+    response = operation.result(90)
 
     # The output is written to GCS with the provided output_uri as prefix
     gcs_output_uri = response.output_config.gcs_destination.uri

diff --git a/samples/snippets/detect/vision_async_batch_annotate_images_test.py b/samples/snippets/detect/vision_async_batch_annotate_images_test.py
@@ -49,6 +49,7 @@ def bucket(storage_client):
         blob.delete()
 
 
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_sample_asyn_batch_annotate_images(storage_client, bucket, capsys):
     input_image_uri = os.path.join(GCS_ROOT, "label/wakeupcat.jpg")