diff --git a/packages/google-cloud-vision/samples/snippets/detect/detect_test.py b/packages/google-cloud-vision/samples/snippets/detect/detect_test.py index a618769df19f..7735733f3d9f 100644 --- a/packages/google-cloud-vision/samples/snippets/detect/detect_test.py +++ b/packages/google-cloud-vision/samples/snippets/detect/detect_test.py @@ -17,6 +17,7 @@ import backoff from google.cloud import storage +import pytest import detect @@ -242,6 +243,7 @@ def test_detect_crop_hints_uri(capsys): assert 'bounds: ' in out +@pytest.mark.flaky(max_runs=3, min_passes=1) def test_async_detect_document(capsys): storage_client = storage.Client() bucket = storage_client.get_bucket(BUCKET)