fix(api): remove any from device pool

ssube · Feb 11, 2023 · db06f4a · db06f4a
1 parent 6b30ccc
commit db06f4a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -400,7 +400,8 @@ def any_first_cpu_last(a: DeviceParams, b: DeviceParams):
 app = Flask(__name__)
 CORS(app, origins=context.cors_origin)
 
-executor = DevicePoolExecutor(available_platforms)
+# any is a fake device, should not be in the pool
+executor = DevicePoolExecutor([p for p in available_platforms if p.device != "any"])
 
 if is_debug():
     gc.set_debug(gc.DEBUG_STATS)