From e052578a20d24537293f7375545d7564f7fda112 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 9 Dec 2023 23:03:41 -0600 Subject: [PATCH] lint, give civitai client a name --- api/onnx_web/convert/client/civitai.py | 2 ++ api/onnx_web/convert/client/http.py | 1 - api/onnx_web/convert/client/huggingface.py | 1 - api/onnx_web/convert/utils.py | 2 +- 4 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/convert/client/civitai.py b/api/onnx_web/convert/client/civitai.py index 706e5c34..1ae5ecfd 100644 --- a/api/onnx_web/convert/client/civitai.py +++ b/api/onnx_web/convert/client/civitai.py @@ -16,7 +16,9 @@ CIVITAI_ROOT = "https://civitai.com/api/download/models/%s" class CivitaiClient(BaseClient): + name = "civitai" protocol = "civitai://" + root: str token: Optional[str] diff --git a/api/onnx_web/convert/client/http.py b/api/onnx_web/convert/client/http.py index 90a5e659..591e3a6c 100644 --- a/api/onnx_web/convert/client/http.py +++ b/api/onnx_web/convert/client/http.py @@ -6,7 +6,6 @@ from ..utils import ( build_cache_paths, download_progress, get_first_exists, - remove_prefix, ) from .base import BaseClient diff --git a/api/onnx_web/convert/client/huggingface.py b/api/onnx_web/convert/client/huggingface.py index 60f9e08f..68a6a853 100644 --- a/api/onnx_web/convert/client/huggingface.py +++ b/api/onnx_web/convert/client/huggingface.py @@ -1,7 +1,6 @@ from logging import getLogger from typing import Optional -from huggingface_hub import snapshot_download from huggingface_hub.file_download import hf_hub_download from ..utils import ConversionContext, remove_prefix diff --git a/api/onnx_web/convert/utils.py b/api/onnx_web/convert/utils.py index 015b13cd..7785007b 100644 --- a/api/onnx_web/convert/utils.py +++ b/api/onnx_web/convert/utils.py @@ -384,7 +384,7 @@ def build_cache_paths( if format is not None: basename = path.basename(name) _filename, ext = path.splitext(basename) - if ext is None or ext == '': + if ext is None or ext == "": name = f"{name}.{format}" paths = [