From 91a141c8b8cd8b51b96061f2bcb7f1f15c8c9ab7 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 9 Dec 2025 07:55:43 +0200 Subject: [PATCH] minor cleanup --- nodes/curve_nodes.py | 4 +--- nodes/mask_nodes.py | 1 + nodes/nodes.py | 1 - 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/nodes/curve_nodes.py b/nodes/curve_nodes.py index 77be51f..053d8c6 100644 --- a/nodes/curve_nodes.py +++ b/nodes/curve_nodes.py @@ -1,14 +1,12 @@ import torch from torchvision import transforms import json -from PIL import Image, ImageDraw, ImageFont, ImageColor, ImageFilter, ImageChops +from PIL import Image, ImageDraw, ImageFont, ImageFilter import numpy as np from ..utility.utility import pil2tensor, tensor2pil import folder_paths import io import base64 - -from comfy.utils import common_upscale def parse_color(color): if isinstance(color, str) and ',' in color: diff --git a/nodes/mask_nodes.py b/nodes/mask_nodes.py index d9d6ad8..cd9db2d 100644 --- a/nodes/mask_nodes.py +++ b/nodes/mask_nodes.py @@ -881,6 +881,7 @@ class CreateVoronoiMask: def createvoronoi(self, frames, num_points, line_width, speed, frame_width, frame_height): from scipy.spatial import Voronoi + from matplotlib import pyplot as plt # Define the number of images in the batch batch_size = frames out = [] diff --git a/nodes/nodes.py b/nodes/nodes.py index da6a527..7a88a7f 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -3,7 +3,6 @@ import torch.nn as nn import numpy as np from PIL import Image import json, re, os, io, time -import re import importlib from comfy import model_management