1
0
Fork 0

apply lint

This commit is contained in:
Sean Sube 2023-07-04 12:09:46 -05:00
parent 6ec7777f77
commit d1ecef9f2e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 6 additions and 4 deletions

View File

@ -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

View File

@ -186,7 +186,6 @@ class ImageParams:
steps: int
seed: int
negative_prompt: Optional[str]
lpw: bool
eta: float
batch: int
control: Optional[NetworkModel]

View File

@ -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]