fix: time out hub push

This commit is contained in:
Florian Hönicke
2023-04-20 18:12:19 +02:00
parent dfdfd6e881
commit cd843f5d3a
3 changed files with 16 additions and 1 deletions

View File

@@ -11,6 +11,7 @@ from pathlib import Path
import click
import hubble
import requests
import timeout_decorator
from hubble.executor.helper import upload_file, archive_package, get_request_header
from jcloud.flow import CloudFlow
from jina import Flow
@@ -62,6 +63,18 @@ In this case, please cancel this run, rerun your gptdeploy command and login int
def push_executor(dir_path):
for i in range(3):
try:
_push_executor(dir_path)
break
except timeout_decorator.timeout_decorator.TimeoutError:
print('timeout error, retrying...')
if i == 2:
raise Exception('timeout error, please try again later')
continue
@timeout_decorator.timeout(10*60)
def _push_executor(dir_path):
dir_path = Path(dir_path)
md5_hash = hashlib.md5()