Skip to content

Commit

Permalink
fix(api): correct file extension validity check
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Nov 17, 2023
1 parent eb3f147 commit c8dd85e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion api/onnx_web/convert/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]:
Expand Down

0 comments on commit c8dd85e

Please sign in to comment.