FIX AutoDL调度修复暖池功能
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import time
|
||||
import traceback
|
||||
import uuid
|
||||
from concurrent.futures.thread import ThreadPoolExecutor
|
||||
from typing import Union
|
||||
@@ -18,6 +19,8 @@ class RunningPool:
|
||||
def run(self, instance_id, uid, base_url:str, video_file:Union[UploadFile,str], audio_file:Union[UploadFile,str]) -> Union[bool, None]:
|
||||
try:
|
||||
code = self._req(base_url, video_file, audio_file)
|
||||
if not code:
|
||||
raise Exception("[Code] is None")
|
||||
except Exception as e:
|
||||
loguru.logger.error("%s Task Submit Failed: %s" % (uid, e))
|
||||
return None
|
||||
@@ -41,6 +44,7 @@ class RunningPool:
|
||||
return {"status": resp.json()['status'], "msg": resp.json()['msg'],
|
||||
"instance_id": self.tasks[uid]["instance_id"]}
|
||||
except Exception as e:
|
||||
traceback.print_exc()
|
||||
loguru.logger.error("%s Get Result Failed: %s" % (uid, e))
|
||||
return None
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user