cancel jobs correctly

2023-02-04 10:10:40 -06:00 · 2023-02-04 10:10:40 -06:00 · 1491a9e1e0
parent c3b0c5e3e3
commit 1491a9e1e0
1 changed files with 1 additions and 2 deletions
--- a/api/onnx_web/device_pool.py
+++ b/api/onnx_web/device_pool.py
@ -92,8 +92,7 @@ class DevicePoolExecutor:
                if job.future.cancel():
                    return True
                else:
-                    with job.cancel.get_lock():
-                        job.cancel.value = True
+                    job.set_cancel()

    def done(self, key: str) -> bool:
        for job in self.jobs: