From 22e05979169a1a52a4593b809d5755739763e526 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 9 Dec 2023 23:09:00 -0600 Subject: [PATCH] pass kwargs on to client --- api/onnx_web/convert/client/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/convert/client/__init__.py b/api/onnx_web/convert/client/__init__.py index 166905e6..23a71177 100644 --- a/api/onnx_web/convert/client/__init__.py +++ b/api/onnx_web/convert/client/__init__.py @@ -35,6 +35,7 @@ def fetch_model( source: str, format: Optional[str] = None, dest: Optional[str] = None, + **kwargs, ) -> str: # TODO: switch to urlparse's default scheme if source.startswith(path.sep) or source.startswith("."): @@ -45,7 +46,7 @@ def fetch_model( if source.startswith(proto): # TODO: fix type of client_type client: BaseClient = client_type() - return client.download(conversion, name, source, format=format, dest=dest) + return client.download(conversion, name, source, format=format, dest=dest, **kwargs) logger.warning("unknown model protocol, using path as provided: %s", source) return source