Skip to content

Commit

Permalink
[vision] fix: add timeout for LRO result and mark it as flaky [(#3684)](
Browse files Browse the repository at this point in the history
  • Loading branch information
Takashi Matsuo authored May 4, 2020
1 parent fbe33c4 commit 974370a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down

0 comments on commit 974370a

Please sign in to comment.