chore(deps): update dependency backoff to v2 (#383)

* chore(deps): update dependency backoff to v2 * test: backoff wait_gen arg expects a generator Co-authored-by: Anthonios Partheniou <partheniou@google.com>
GoogleCloudPlatform · Apr 26, 2022 · ab40e26 · ab40e26
1 parent 683d2de
commit ab40e26
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/automl/snippets/requirements-test.txt b/automl/snippets/requirements-test.txt
@@ -1,2 +1,2 @@
-backoff==1.11.1
+backoff==2.0.0
 pytest==7.1.2
diff --git a/automl/tables/predict_test.py b/automl/tables/predict_test.py
@@ -31,7 +31,9 @@
 
 
 @backoff.on_exception(
-    wait_gen=lambda: iter([50, 150, MAX_TIMEOUT]), exception=Exception, max_tries=3
+    wait_gen=lambda: (wait_time for wait_time in [50, 150, MAX_TIMEOUT]),
+    exception=Exception,
+    max_tries=3,
 )
 def test_predict(capsys):
     inputs = {

diff --git a/automl/tables/requirements-test.txt b/automl/tables/requirements-test.txt
@@ -1,2 +1,2 @@
 pytest==7.1.2
-backoff==1.11.1
+backoff==2.0.0