From c8dd85e798564bd0e9bc6f442acaa174c584f098 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 16 Nov 2023 22:04:11 -0600 Subject: [PATCH] fix(api): correct file extension validity check --- api/onnx_web/convert/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/convert/utils.py b/api/onnx_web/convert/utils.py index 42cb73703..61b6547ff 100644 --- a/api/onnx_web/convert/utils.py +++ b/api/onnx_web/convert/utils.py @@ -192,7 +192,7 @@ def check_ext(name: str, exts: List[str]) -> Tuple[bool, str]: _name, ext = path.splitext(name) ext = ext.strip(".") - return (name in exts, ext) + return (ext in exts, ext) def source_format(model: Dict) -> Optional[str]: