chore: fix flaky test by adding rerun (#77)

* chore: fix flaky test by adding rerun * fixed the lint
GoogleCloudPlatform · Nov 15, 2022 · c6d8094 · c6d8094
1 parent d759ee0
commit c6d8094
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vision/snippets/detect/detect_test.py b/vision/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)