From 3ae7ecfe828a9e54d20c1102c0b14511b414dcfc Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Tue, 22 Apr 2025 10:20:50 -0700 Subject: [PATCH] Ruff. --- comfy_api_nodes/nodes_api.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/comfy_api_nodes/nodes_api.py b/comfy_api_nodes/nodes_api.py index 07728a91e..b8a361470 100644 --- a/comfy_api_nodes/nodes_api.py +++ b/comfy_api_nodes/nodes_api.py @@ -100,7 +100,6 @@ class IdeogramTextToImage(ComfyNodeABC): def api_call(self, prompt, model, aspect_ratio=None, resolution=None, magic_prompt_option="AUTO", seed=0, style_type="NONE", negative_prompt="", num_images=1, color_palette="", auth_token=None): - import io import numpy as np import requests @@ -237,13 +236,12 @@ class OpenAITextToImage(ComfyNodeABC): elif model == "dall-e-3": if size == "auto": size = "1024x1024" - valid_sizes = ["1024x1024", "1792x1024", "1024x1792"] + valid_sizes = ["1024x1024", "1792x1024", "1024x1792"] if size not in valid_sizes: raise ValueError(f"Size {size} not valid for dall-e-3. Must be one of: {', '.join(valid_sizes)}") # TODO: add NEW MODEL - import io import numpy as np import torch @@ -271,7 +269,7 @@ class OpenAITextToImage(ComfyNodeABC): response = operation.execute() # validate raw JSON response - + data = response.data if not data or len(data) == 0: raise Exception("No images returned from OpenAI endpoint")