From 0cf64ae41155be660ebcf94011e4f30e89d705f6 Mon Sep 17 00:00:00 2001 From: Vito Sansevero Date: Sat, 14 Jun 2025 15:25:25 -0700 Subject: [PATCH] style(logic): adjust typing imports and line breaks --- kikotools/tools/resolution_calculator/logic.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/kikotools/tools/resolution_calculator/logic.py b/kikotools/tools/resolution_calculator/logic.py index d89e8db..9928605 100644 --- a/kikotools/tools/resolution_calculator/logic.py +++ b/kikotools/tools/resolution_calculator/logic.py @@ -4,7 +4,7 @@ Pure functions for dimension extraction and scaling calculations """ import torch -from typing import Tuple, Optional, Union, Dict, Any +from typing import Tuple, Optional, Dict def extract_dimensions( @@ -16,7 +16,8 @@ def extract_dimensions( Args: image: Optional IMAGE tensor in ComfyUI format [batch, height, width, channels] - latent: Optional LATENT dict with 'samples' tensor [batch, channels, height/8, width/8] + latent: Optional LATENT dict with 'samples' tensor + [batch, channels, height/8, width/8] Returns: Tuple of (width, height) as integers @@ -42,7 +43,8 @@ def extract_dimensions( samples = latent["samples"] if len(samples.shape) != 4: raise ValueError( - f"Expected LATENT samples tensor with 4 dimensions, got {len(samples.shape)}" + f"Expected LATENT samples tensor with 4 dimensions, " + f"got {len(samples.shape)}" ) _, _, latent_height, latent_width = samples.shape