fix: Improve flaky retry after periodic test suite failure (GoogleCloudPlatform#10068)

nicain · engelke · web-flow · commit 9685dce914f6 · 2023-05-26T16:46:00.000-07:00
Co-authored-by: Charles Engelke &lt;engelke@google.com&gt;
diff --git a/speech/snippets/adaptation_v2_phrase_set_reference_test.py b/speech/snippets/adaptation_v2_phrase_set_reference_test.py
@@ -16,7 +16,7 @@
 import re
 from uuid import uuid4
 
-from google.api_core.retry import Retry
+import backoff
 from google.cloud.speech_v2 import SpeechClient
 from google.cloud.speech_v2.types import cloud_speech
 
@@ -37,7 +37,7 @@ def delete_phrase_set(name):
     client.delete_phrase_set(request=request)
 
 
-@Retry()
+@backoff.on_exception(backoff.expo, Exception, max_time=120)
 def test_adaptation_v2_phrase_set_reference(capsys):
     project_id = os.getenv("GOOGLE_CLOUD_PROJECT")
 
diff --git a/speech/snippets/requirements-test.txt b/speech/snippets/requirements-test.txt
@@ -1,2 +1,3 @@
+backoff==2.2.1
 flaky==3.7.0
 pytest==7.2.2
