From 2449e347f0895284b7930133f7d32186bb2ed0fe Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 17 Feb 2024 15:24:47 -0600 Subject: [PATCH] make optional params properly optional --- api/onnx_web/server/params.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/server/params.py b/api/onnx_web/server/params.py index c2d212b9..6dc36726 100644 --- a/api/onnx_web/server/params.py +++ b/api/onnx_web/server/params.py @@ -473,10 +473,22 @@ def get_request_params( data = get_request_data() device, params, size = pipeline_from_json(server, data, default_pipeline) - border = build_border(data["border"]) - upscale = build_upscale(data["upscale"]) - highres = build_highres(data["highres"]) - experimental = build_experimental(data["experimental"]) + + border = None + if "border" in data: + border = build_border(data["border"]) + + upscale = None + if "upscale" in data: + upscale = build_upscale(data["upscale"]) + + highres = None + if "highres" in data: + highres = build_highres(data["highres"]) + + experimental = None + if "experimental" in data: + experimental = build_experimental(data["experimental"]) return RequestParams( device,