fix possible overflow in color_mask

MeetWq · Aug 17, 2024 · a6e9796 · a6e9796
1 parent 8c0744a
commit a6e9796
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pil_utils/build_image.py b/pil_utils/build_image.py
@@ -335,7 +335,7 @@ def color_mask(self, color: ColorType) -> "BuildImage":
         rgb_sum = sum(color)
         for i in range(h):
             for j in range(w):
-                value = img_gray[i, j]
+                value = int(img_gray[i, j])
                 new_color = (
                     [
                         int(value * r / rgb_sum),

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pil-utils"
-version = "0.1.11"
+version = "0.1.12"
 description = "A simple PIL wrapper and text-to-image tool"
 authors = ["meetwq <[email protected]>"]
 license = "MIT"