Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-01 23:08:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

gdrive: do not import pydrive2 on module load (#3399)

Browse Source
This commit is contained in:
Alexander Schepanovski
2020-02-25 14:52:40 +02:00
committed by GitHub
parent 8d04d6d3ce
commit a81ab5e26b
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/remote/gdrive.py
+6 -4
View File
@@ -39,16 +39,18 @@ class GDriveMissedCredentialKeyError(DvcException):
def gdrive_retry(func):
from pydrive2.files import ApiRequestError
retry_re = re.compile(r"HttpError (403|500|502|503|504)")
def should_retry(exc):
from pydrive2.files import ApiRequestError
return isinstance(exc, ApiRequestError) and retry_re.search(str(exc))
# 15 tries, start at 0.5s, multiply by golden ratio, cap at 20s
return retry(
15,
timeout=lambda a: min(0.5 * 1.618 ** a, 20),
errors=ApiRequestError,
filter_errors=lambda exc: retry_re.search(str(exc)),
filter_errors=should_retry,
)(func)