Add shuffle to Data List and LIST

Bump version to 1.3.0
This commit is contained in:
StableLlama
2026-01-31 19:42:33 +01:00
parent 607093e9cd
commit 443e75be09
6 changed files with 141 additions and 3 deletions
@@ -920,6 +920,39 @@ class DataListSetItem(ComfyNodeABC):
raise IndexError(f"Index {index} out of range for list of length {len(input_list)}")
class DataListShuffle(ComfyNodeABC):
"""
Shuffles the items in a list using a seed for reproducibility.
This node takes a list and a seed as input and returns a new shuffled list.
"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"list": (IO.ANY, {}),
"seed": (IO.INT, {"default": 0}),
}
}
RETURN_TYPES = (IO.ANY,)
RETURN_NAMES = ("list",)
CATEGORY = "Basic/Data List"
DESCRIPTION = cleandoc(__doc__ or "")
FUNCTION = "shuffle_list"
INPUT_IS_LIST = True
OUTPUT_IS_LIST = (True,)
def shuffle_list(self, **kwargs: list[Any]) -> tuple[list[Any]]:
import random
input_list = kwargs.get('list', [])
seed = kwargs.get('seed', [0])[0]
random.seed(seed)
result = input_list.copy()
random.shuffle(result)
return (result,)
class DataListSlice(ComfyNodeABC):
"""
Creates a slice of a list.
@@ -1145,6 +1178,7 @@ NODE_CLASS_MAPPINGS = {
"Basic data handling: DataListRemove": DataListRemove,
"Basic data handling: DataListReverse": DataListReverse,
"Basic data handling: DataListSetItem": DataListSetItem,
"Basic data handling: DataListShuffle": DataListShuffle,
"Basic data handling: DataListSlice": DataListSlice,
"Basic data handling: DataListSort": DataListSort,
"Basic data handling: DataListSum": DataListSum,
@@ -1183,6 +1217,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Basic data handling: DataListRemove": "remove",
"Basic data handling: DataListReverse": "reverse",
"Basic data handling: DataListSetItem": "set item",
"Basic data handling: DataListShuffle": "shuffle",
"Basic data handling: DataListSlice": "slice",
"Basic data handling: DataListSort": "sort",
"Basic data handling: DataListSum": "sum",
+31
View File
@@ -741,6 +741,35 @@ class ListSetItem(ComfyNodeABC):
except IndexError:
raise IndexError(f"Index {index} out of range for LIST of length {len(list)}")
class ListShuffle(ComfyNodeABC):
"""
Shuffles the items in a list using a seed for reproducibility.
This node takes a LIST and a seed as input and returns a new shuffled LIST.
"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"list": ("LIST", {}),
"seed": ("INT", {"default": 0}),
}
}
RETURN_TYPES = ("LIST",)
CATEGORY = "Basic/LIST"
DESCRIPTION = cleandoc(__doc__ or "")
FUNCTION = "shuffle_list"
def shuffle_list(self, list: list[Any], seed: int) -> tuple[list[Any]]:
import random
random.seed(seed)
result = list.copy()
random.shuffle(result)
return (result,)
class ListSlice(ComfyNodeABC):
"""
Creates a slice of a LIST.
@@ -910,6 +939,7 @@ NODE_CLASS_MAPPINGS = {
"Basic data handling: ListRemove": ListRemove,
"Basic data handling: ListReverse": ListReverse,
"Basic data handling: ListSetItem": ListSetItem,
"Basic data handling: ListShuffle": ListShuffle,
"Basic data handling: ListSlice": ListSlice,
"Basic data handling: ListSort": ListSort,
"Basic data handling: ListSum": ListSum,
@@ -944,6 +974,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Basic data handling: ListRemove": "remove",
"Basic data handling: ListReverse": "reverse",
"Basic data handling: ListSetItem": "set item",
"Basic data handling: ListShuffle": "shuffle",
"Basic data handling: ListSlice": "slice",
"Basic data handling: ListSort": "sort",
"Basic data handling: ListSum": "sum",