samples: fixed flaky test on vision beta samples (#62)

GoogleCloudPlatform · Oct 20, 2020 · f0b583c · f0b583c
1 parent 9f6a844
commit f0b583c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/videointelligence/samples/analyze/beta_snippets_test.py b/videointelligence/samples/analyze/beta_snippets_test.py
@@ -112,6 +112,7 @@ def test_detect_explicit_content_streaming(capsys, video_path):
 
 
 @pytest.mark.slow
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_annotation_to_storage_streaming(capsys, video_path, bucket):
     output_uri = "gs://{}".format(bucket.name)
     beta_snippets.annotation_to_storage_streaming(video_path, output_uri)