lint updates

pull/2878/head
Vladimir Mandic 2024-02-16 15:09:22 -05:00
parent 1b04494cec
commit 8caa82204c
3 changed files with 8 additions and 7 deletions

View File

@ -54,17 +54,17 @@ def pil_to_temp_file(self, img: Image, dir: str, format="png") -> str: # pylint:
debug(f'Image registered: {name}')
return name
if shared.opts.temp_dir != "":
dir = shared.opts.temp_dir
folder = shared.opts.temp_dir
use_metadata = False
metadata = PngImagePlugin.PngInfo()
for key, value in img.info.items():
if isinstance(key, str) and isinstance(value, str):
metadata.add_text(key, value)
use_metadata = True
if not os.path.exists(dir):
os.makedirs(dir, exist_ok=True)
shared.log.debug(f'Created temp folder: path="{dir}"')
with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir=dir) as tmp:
if not os.path.exists(folder):
os.makedirs(folder, exist_ok=True)
shared.log.debug(f'Created temp folder: path="{folder}"')
with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir=folder) as tmp:
name = tmp.name
img.save(name, pnginfo=(metadata if use_metadata else None))
img.already_saved_as = name

View File

@ -10,7 +10,8 @@ exclude = [
"repositories/taming",
"repositories/blip",
"repositories/codeformer",
"modules/control/proc/normalbae/nets/submodules/efficientnet_repo/geffnet",
"modules/control/proc/normalbae/",
"modules/control/proc/leres/",
"modules/control/units/*_model.py",
"modules/control/units/*_pipe.py",
"modules/pipelines/*.py",

2
wiki

@ -1 +1 @@
Subproject commit eaa8ca8f42d8031fdb796ea01845201fa2b75e12
Subproject commit 569aa7a72fdac20e4136333cfb66a27f63257a38