chore: another flaky test for video beta samples (#91)

* chore: another flaky test for video beta samples * changed it to timeout
googleapis · Dec 21, 2020 · ac8b1c3 · ac8b1c3
1 parent ba875f2
commit ac8b1c3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/google-cloud-videointelligence/samples/analyze/beta_snippets.py b/packages/google-cloud-videointelligence/samples/analyze/beta_snippets.py
@@ -45,7 +45,7 @@
 import io
 
 
-def speech_transcription(input_uri):
+def speech_transcription(input_uri, timeout=180):
     # [START video_speech_transcription_gcs_beta]
     """Transcribe speech from a video stored on GCS."""
     from google.cloud import videointelligence_v1p1beta1 as videointelligence
@@ -69,7 +69,7 @@ def speech_transcription(input_uri):
 
     print("\nProcessing video for speech transcription.")
 
-    result = operation.result(timeout=180)
+    result = operation.result(timeout)
 
     # There is only one annotation_result since only
     # one video is processed.

diff --git a/packages/google-cloud-videointelligence/samples/analyze/beta_snippets_test.py b/packages/google-cloud-videointelligence/samples/analyze/beta_snippets_test.py
@@ -73,7 +73,7 @@ def delete_bucket():
 @pytest.mark.slow
 def test_speech_transcription(capsys):
     beta_snippets.speech_transcription(
-        "gs://python-docs-samples-tests/video/googlework_short.mp4"
+        "gs://python-docs-samples-tests/video/googlework_short.mp4", timeout=240
     )
     out, _ = capsys.readouterr()
     assert "cultural" in out