Skip to content

Commit fb9ef51

Browse files
author
Takashi Matsuo
authored
[vision] fix: longer timeout (#3447)
fixes #2962
1 parent 9c492b5 commit fb9ef51

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

vision/cloud-client/detect/detect.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -836,7 +836,7 @@ def async_detect_document(gcs_source_uri, gcs_destination_uri):
836836
requests=[async_request])
837837

838838
print('Waiting for the operation to finish.')
839-
operation.result(timeout=180)
839+
operation.result(timeout=300)
840840

841841
# Once the request has completed and the output has been
842842
# written to GCS, we can list all the output files.

vision/cloud-client/detect/detect_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import uuid
1717

1818
from google.cloud import storage
19+
import pytest
1920

2021
import detect
2122

@@ -208,6 +209,7 @@ def test_detect_crop_hints_uri(capsys):
208209
assert 'bounds: ' in out
209210

210211

212+
@pytest.mark.flaky
211213
def test_async_detect_document(capsys):
212214
storage_client = storage.Client()
213215
bucket = storage_client.get_bucket(BUCKET)
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
pytest==5.3.2
2+
flaky==3.6.1

0 commit comments

Comments
 (0)