pass kwargs on to client
This commit is contained in:
parent
35c973e55f
commit
22e0597916
|
@ -35,6 +35,7 @@ def fetch_model(
|
||||||
source: str,
|
source: str,
|
||||||
format: Optional[str] = None,
|
format: Optional[str] = None,
|
||||||
dest: Optional[str] = None,
|
dest: Optional[str] = None,
|
||||||
|
**kwargs,
|
||||||
) -> str:
|
) -> str:
|
||||||
# TODO: switch to urlparse's default scheme
|
# TODO: switch to urlparse's default scheme
|
||||||
if source.startswith(path.sep) or source.startswith("."):
|
if source.startswith(path.sep) or source.startswith("."):
|
||||||
|
@ -45,7 +46,7 @@ def fetch_model(
|
||||||
if source.startswith(proto):
|
if source.startswith(proto):
|
||||||
# TODO: fix type of client_type
|
# TODO: fix type of client_type
|
||||||
client: BaseClient = 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)
|
logger.warning("unknown model protocol, using path as provided: %s", source)
|
||||||
return source
|
return source
|
||||||
|
|
Loading…
Reference in New Issue