diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index c6ea707e..1926d481 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -12,6 +12,12 @@ logger = getLogger(__name__) class JobContext: + cancel: Value[bool] = None + device_index: Value[int] = None + devices: List[DeviceParams] = None + key: str = None + progress: Value[int] = None + def __init__( self, key: str, @@ -70,6 +76,10 @@ class Job: Link a future to its context. ''' + context: JobContext = None + future: Future = None + key: str = None + def __init__( self, key: str,