apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add value error handling

This commit is contained in:
Sean Sube 2023-02-27 23:12:53 -06:00
parent f7f438e767
commit 1ce98ace33
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/worker/pool.py
View File

@ -95,6 +95,8 @@ class DevicePoolExecutor:
f.write(str(job) + "\n\n")
except Empty:
pass
except ValueError:
break
except Exception as err:
logger.error("error in log worker: %s", err)
@ -119,6 +121,8 @@ class DevicePoolExecutor:
self.context[device].set_cancel()
except Empty:
pass
except ValueError:
break
except Exception as err:
logger.error("error in progress worker: %s", err)
@ -141,6 +145,8 @@ class DevicePoolExecutor:
del self.active_jobs[job]
except Empty:
pass
except ValueError:
break
except Exception as err:
logger.error("error in finished worker: %s", err)
@ -219,6 +225,7 @@ class DevicePoolExecutor:
if worker.is_alive():
logger.debug("stopping worker for device %s", device)
worker.join(self.join_timeout)
worker.terminate()
else:
logger.debug("worker for device %s has died", device)

5
api/onnx_web/worker/worker.py
View File

@ -29,11 +29,16 @@ def worker_main(context: WorkerContext, server: ServerContext):
logger.info("starting job: %s", name)
fn(context, *args, **kwargs)
logger.info("job succeeded: %s", name)
context.pending.task_done()
context.set_finished()
except Empty:
pass
except KeyboardInterrupt:
logger.info("worker got keyboard interrupt")
exit(0)
except ValueError as e:
logger.info("value error in worker: %s", e)
exit(1)
except Exception as e:
logger.error(
"error while running job: %s",