From 9c11b8e33e9fef45ed79a7616a3a052d52851174 Mon Sep 17 00:00:00 2001
From: Mehdi Chaid <JeyzerMC@users.noreply.github.com>
Date: Thu, 14 May 2026 11:13:10 -0400
Subject: [PATCH] fix: use valid string types for sampler/scheduler in
 _ALL_PINS Fix FlakeDataSplitSelect not creatable: _ALL_PINS used
 _SAMPLER_TYPES[0] and [1] which are tuples/lists when comfy.samplers imports
 successfully, not valid RETURN_TYPES strings. Use plain 'SAMPLER'/'SCHEDULER'
 type names instead.

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

diff --git a/flake_data_nodes.py b/flake_data_nodes.py
index f3d6553..93a6dff 100644
--- a/flake_data_nodes.py
+++ b/flake_data_nodes.py
@@ -24,8 +24,8 @@ _ALL_PINS = [
     ("height", "INT"),
     ("steps", "INT"),
     ("cfg", "FLOAT"),
-    ("sampler_name", _SAMPLER_TYPES[0]),
-    ("scheduler", _SAMPLER_TYPES[1]),
+    ("sampler_name", "SAMPLER"),
+    ("scheduler", "SCHEDULER"),
 ]