From d75cd50a62e153bd32827b9b02b48149f97b1c8a Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 16:57:00 -0600 Subject: [PATCH] fix non-generic types --- api/onnx_web/device_pool.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 1926d481..558185b3 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -2,7 +2,7 @@ from collections import Counter from concurrent.futures import Future, ThreadPoolExecutor, ProcessPoolExecutor from logging import getLogger from multiprocessing import Value -from typing import Any, Callable, Dict, List, Optional, Tuple, Union +from typing import Any, Callable, List, Optional, Tuple, Union from .params import ( DeviceParams, @@ -12,11 +12,11 @@ logger = getLogger(__name__) class JobContext: - cancel: Value[bool] = None - device_index: Value[int] = None + cancel: Value = None + device_index: Value = None devices: List[DeviceParams] = None key: str = None - progress: Value[int] = None + progress: Value = None def __init__( self,