chore: increased timeout for flaky service & added retry (#103)

* chore: increased timeout for flaky service & added retry * fixed param error * lint
GoogleCloudPlatform · Sep 15, 2023 · 79cfb2f · 79cfb2f
1 parent 4f93a72
commit 79cfb2f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/videointelligence/samples/analyze/beta_snippets.py b/videointelligence/samples/analyze/beta_snippets.py
@@ -510,7 +510,7 @@ def stream_generator():
     # The default timeout is about 300 seconds.
     # To process longer videos it should be set to
     # larger than the length (in seconds) of the stream.
-    responses = client.streaming_annotate_video(requests, timeout=600)
+    responses = client.streaming_annotate_video(requests, timeout=900)
 
     # Each response corresponds to about 1 second of video.
     for response in responses:

diff --git a/videointelligence/samples/analyze/beta_snippets_test.py b/videointelligence/samples/analyze/beta_snippets_test.py
@@ -95,7 +95,9 @@ def test_detect_shot_change_streaming(capsys, video_path):
     assert "Shot" in out
 
 
+# Flaky ServiceUnavailable
 @pytest.mark.slow
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_track_objects_streaming(capsys, video_path):
     beta_snippets.track_objects_streaming(video_path)