From 15ecaa7370ed1a498c26b10fd57ef940dbee696c Mon Sep 17 00:00:00 2001 From: filipstrand Date: Sun, 22 Sep 2024 21:06:42 +0200 Subject: [PATCH] Rename util method 'save' to 'save_image' --- src/mflux/controlnet/controlnet_util.py | 2 +- src/mflux/post_processing/generated_image.py | 2 +- src/mflux/post_processing/image_util.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/mflux/controlnet/controlnet_util.py b/src/mflux/controlnet/controlnet_util.py index daacf33..e7069d0 100644 --- a/src/mflux/controlnet/controlnet_util.py +++ b/src/mflux/controlnet/controlnet_util.py @@ -30,4 +30,4 @@ class ControlnetUtil: def save_canny_image(control_image, path: str): base, ext = os.path.splitext(path) new_filename = f"{base}_controlnet_canny{ext}" - ImageUtil.save(control_image, new_filename) + ImageUtil.save_image(control_image, new_filename) diff --git a/src/mflux/post_processing/generated_image.py b/src/mflux/post_processing/generated_image.py index 85a21e4..35019ef 100644 --- a/src/mflux/post_processing/generated_image.py +++ b/src/mflux/post_processing/generated_image.py @@ -39,7 +39,7 @@ class GeneratedImage: self.controlnet_strength = controlnet_strength def save(self, path: str, export_json_metadata: bool = False) -> None: - ImageUtil.save(self.image, path, self._get_metadata(), export_json_metadata) + ImageUtil.save_image(self.image, path, self._get_metadata(), export_json_metadata) def _get_metadata(self) -> dict: return { diff --git a/src/mflux/post_processing/image_util.py b/src/mflux/post_processing/image_util.py index 52f94de..56b4145 100644 --- a/src/mflux/post_processing/image_util.py +++ b/src/mflux/post_processing/image_util.py @@ -87,7 +87,7 @@ class ImageUtil: return Image.open(path) @staticmethod - def save( + def save_image( image: PIL.Image.Image, path: str, metadata: dict | None = None,