From d1ecef9f2e8c026040cb8a8c6845a717116421e9 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 4 Jul 2023 12:09:46 -0500 Subject: [PATCH] apply lint --- api/onnx_web/chain/base.py | 2 +- api/onnx_web/params.py | 1 - api/onnx_web/utils.py | 7 +++++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py index 6d950b5c..1ca1ba5b 100644 --- a/api/onnx_web/chain/base.py +++ b/api/onnx_web/chain/base.py @@ -1,7 +1,7 @@ from datetime import timedelta from logging import getLogger from time import monotonic -from typing import Any, List, Optional, Protocol, Tuple +from typing import Any, List, Optional, Tuple from PIL import Image diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index 84dd1869..9b418c50 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -186,7 +186,6 @@ class ImageParams: steps: int seed: int negative_prompt: Optional[str] - lpw: bool eta: float batch: int control: Optional[NetworkModel] diff --git a/api/onnx_web/utils.py b/api/onnx_web/utils.py index 0c1b115d..18c2ebbd 100644 --- a/api/onnx_web/utils.py +++ b/api/onnx_web/utils.py @@ -6,7 +6,7 @@ from json import JSONDecodeError from logging import getLogger from os import environ, path from platform import system -from typing import Any, Dict, List, Optional, Sequence, Union, TypeVar +from typing import Any, Dict, List, Optional, Sequence, TypeVar, Union import torch from yaml import safe_load @@ -45,6 +45,7 @@ def get_and_clamp_int( TElem = TypeVar("TElem") + def get_from_list( args: Any, key: str, values: Sequence[TElem], default_value: Optional[TElem] = None ) -> Optional[TElem]: @@ -59,7 +60,9 @@ def get_from_list( return None -def get_from_map(args: Any, key: str, values: Dict[str, TElem], default: TElem) -> TElem: +def get_from_map( + args: Any, key: str, values: Dict[str, TElem], default: TElem +) -> TElem: selected = args.get(key, default) if selected in values: return values[selected]