From 65565c1d2123b09400f32d1e3f6fed1abdcc9052 Mon Sep 17 00:00:00 2001 From: catboxanon <122327233+catboxanon@users.noreply.github.com> Date: Thu, 1 May 2025 18:38:43 -0400 Subject: [PATCH] Add type hints --- comfy_extras/nodes_custom_sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py index 758696478..3e5be3d3c 100644 --- a/comfy_extras/nodes_custom_sampler.py +++ b/comfy_extras/nodes_custom_sampler.py @@ -266,7 +266,7 @@ class ExtendIntermediateSigmas: FUNCTION = "extend" - def extend(self, sigmas, steps, start_at_sigma, end_at_sigma, spacing): + def extend(self, sigmas: torch.Tensor, steps: int, start_at_sigma: float, end_at_sigma: float, spacing: str): if start_at_sigma < 0: start_at_sigma = float("inf")