[SDK-302] Remove backoff library (#1255)

Author: vbrodsky

.yapfignore

@@ -1 +1,3 @@
-pytest.ini
+Makefile
+*.txt
+*.ini

labelbox/schema/annotation_import.py

@@ -5,7 +5,7 @@
 import time
 from typing import Any, BinaryIO, Dict, List, Union, TYPE_CHECKING, cast
 
-import backoff
+from google.api_core import retry
 from labelbox import parser
 import requests
 from tqdm import tqdm  # type: ignore
@@ -109,12 +109,9 @@ def wait_until_done(self,
             pbar.update(100 - pbar.n)
             pbar.close()
 
-    @backoff.on_exception(
-        backoff.expo,
-        (labelbox.exceptions.ApiLimitError, labelbox.exceptions.TimeoutError,
-         labelbox.exceptions.NetworkError),
-        max_tries=10,
-        jitter=None)
+    @retry.Retry(predicate=retry.if_exception_type(
+        labelbox.exceptions.ApiLimitError, labelbox.exceptions.TimeoutError,
+        labelbox.exceptions.NetworkError))
     def __backoff_refresh(self) -> None:
         self.refresh()
 

labelbox/schema/bulk_import_request.py

@@ -6,7 +6,7 @@
 import logging
 from pathlib import Path
 import pydantic
-import backoff
+from google.api_core import retry
 from labelbox import parser
 import requests
 from pydantic import BaseModel, root_validator, validator
@@ -197,11 +197,9 @@ def wait_until_done(self, sleep_time_seconds: int = 5) -> None:
             time.sleep(sleep_time_seconds)
             self.__exponential_backoff_refresh()
 
-    @backoff.on_exception(
-        backoff.expo, (lb_exceptions.ApiLimitError, lb_exceptions.TimeoutError,
-                       lb_exceptions.NetworkError),
-        max_tries=10,
-        jitter=None)
+    @retry.Retry(predicate=retry.if_exception_type(lb_exceptions.ApiLimitError,
+                                                   lb_exceptions.TimeoutError,
+                                                   lb_exceptions.NetworkError))
     def __exponential_backoff_refresh(self) -> None:
         self.refresh()
 

mypy.ini

@@ -1,8 +1,5 @@
 [mypy]
 
-[mypy-backoff.*]
-ignore_missing_imports = True
-
 [mypy-google.*]
 ignore_missing_imports = True
 

requirements.txt

@@ -1,4 +1,3 @@
-backoff==1.10.0
 geojson
 google-api-core>=1.22.1
 imagesize

setup.py

@@ -20,8 +20,8 @@
     url="https://labelbox.com",
     packages=setuptools.find_packages(),
     install_requires=[
-        "backoff==1.10.0", "requests>=2.22.0", "google-api-core>=1.22.1",
-        "pydantic>=1.8,<2.0", "tqdm", "python-dateutil>=2.8.2,<2.9.0"
+        "requests>=2.22.0", "google-api-core>=1.22.1", "pydantic>=1.8,<2.0",
+        "tqdm", "python-dateutil>=2.8.2,<2.9.0"
     ],
     extras_require={
         'data': [