From db13215b51b28207c4492d04c993e4c92f4f7184 Mon Sep 17 00:00:00 2001
From: Mike <45373284+munkhuushmgl@users.noreply.github.com>
Date: Wed, 2 Dec 2020 11:48:25 -0800
Subject: [PATCH] chore: fix flaky test by adding rerun (#77)

* chore: fix flaky test by adding rerun

* fixed the lint
---
 vision/snippets/detect/detect_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/vision/snippets/detect/detect_test.py b/vision/snippets/detect/detect_test.py
index a618769df19f..7735733f3d9f 100644
--- 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)