From e07274aab36a53e4e94879e43510fd997fd5011e Mon Sep 17 00:00:00 2001
From: EnragedAntelope <176548309+EnragedAntelope@users.noreply.github.com>
Date: Sat, 8 Nov 2025 08:56:03 -0500
Subject: [PATCH] Refactor ConstrainResolution class and methods

---
 nodes.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 075c1cf..3b30c08 100644
--- a/nodes.py
+++ b/nodes.py
@@ -79,7 +79,8 @@ class ConstrainResolution(io.ComfyNode):
                 ),
 
                 # Constraint behavior
-                ConstraintMode.Input(
+                io.Enum.Input(
+                    ConstraintMode,
                     "constraint_mode",
                     default=ConstraintMode.MIN_RES,
                     tooltip=(
@@ -98,7 +99,8 @@ class ConstrainResolution(io.ComfyNode):
                         "Disable if preserving the entire image is more important than exact dimensions."
                     )
                 ),
-                CropPosition.Input(
+                io.Enum.Input(
+                    CropPosition,
                     "crop_position",
                     default=CropPosition.CENTER,
                     tooltip=(