From d2ee9d24ac4d351207f91360fbc646dea5e398f3 Mon Sep 17 00:00:00 2001
From: MMaker <mmaker@mmaker.moe>
Date: Wed, 15 Nov 2023 20:24:16 -0500
Subject: [PATCH] Formatting fixes

---
 __init__.py                  | 2 +-
 mmaker_color_enhance_core.py | 8 ++++----
 scripts/color_enhance.py     | 4 +---
 3 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/__init__.py b/__init__.py
index 6553283..5d52004 100644
--- a/__init__.py
+++ b/__init__.py
@@ -6,4 +6,4 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "MMakerColorEnhance": "Color Enhance",
-}
\ No newline at end of file
+}
diff --git a/mmaker_color_enhance_core.py b/mmaker_color_enhance_core.py
index 87b0522..e9188bb 100644
--- a/mmaker_color_enhance_core.py
+++ b/mmaker_color_enhance_core.py
@@ -6,9 +6,9 @@ import imageio.core.util
 imageio.core.util._precision_warn = lambda *args, **kwargs: None
 
 def color_enhance(arr, strength: float = 1) -> Image.Image:
-        lch = skimage.color.lab2lch(lab=skimage.color.rgb2lab(rgb=np.array(arr, dtype=np.uint8)))
-        lch[:, :, 1] *= 100/(lerp(100, lch[:, :, 1].max(), strength)) # Normalize chroma component
-        return Image.fromarray(np.array(skimage.color.lab2rgb(lab=skimage.color.lch2lab(lch=lch)) * 255, dtype=np.uint8))
+    lch = skimage.color.lab2lch(lab=skimage.color.rgb2lab(rgb=np.array(arr, dtype=np.uint8)))
+    lch[:, :, 1] *= 100/(lerp(100, lch[:, :, 1].max(), strength)) # Normalize chroma component
+    return Image.fromarray(np.array(skimage.color.lab2rgb(lab=skimage.color.lch2lab(lch=lch)) * 255, dtype=np.uint8))
 
 def lerp(a: float, b: float, t: float) -> float:
-        return (1 - t) * a + t * b
\ No newline at end of file
+    return (1 - t) * a + t * b
diff --git a/scripts/color_enhance.py b/scripts/color_enhance.py
index 9d920b6..8931a1e 100644
--- a/scripts/color_enhance.py
+++ b/scripts/color_enhance.py
@@ -17,10 +17,8 @@ class ScriptPostprocessingColorEnhance(scripts_postprocessing.ScriptPostprocessi
     def process(self, pp: scripts_postprocessing.PostprocessedImage, strength):
         if strength == 0:
             return
-        
+
         info_bak = {} if not hasattr(pp.image, "info") else pp.image.info
         pp.image = lib.color_enhance(pp.image, strength)
         pp.image.info = info_bak
         pp.info["Color Enhance"] = strength
-
-    
\ No newline at end of file