diff --git a/gates/bucket_node.py b/gates/bucket_node.py index 5d57f21..c682129 100644 --- a/gates/bucket_node.py +++ b/gates/bucket_node.py @@ -39,7 +39,7 @@ def fit_mask(mask, W, H): class BucketResize: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("IMAGE", "MASK", "INT", "INT", "STRING") RETURN_NAMES = ("image", "mask", "width", "height", "label") diff --git a/gates/gate.py b/gates/gate.py index ae62b52..f32a034 100644 --- a/gates/gate.py +++ b/gates/gate.py @@ -25,7 +25,7 @@ def mask_from_stash(data, image): class ImageGate: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("MASK",) + ("IMAGE",) * MAX_ROUTES RETURN_NAMES = ("mask",) + tuple(f"route_{i + 1}" for i in range(MAX_ROUTES)) diff --git a/gates/loader.py b/gates/loader.py index 9ed70aa..4190f42 100644 --- a/gates/loader.py +++ b/gates/loader.py @@ -27,7 +27,7 @@ def load_image_and_mask(path): class FolderImageLoader: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("IMAGE", "STRING", "MASK", "STRING", "INT") RETURN_NAMES = ("image", "text", "mask", "filename", "index") diff --git a/gates/node.py b/gates/node.py index 9dc1bba..263e9f0 100644 --- a/gates/node.py +++ b/gates/node.py @@ -8,7 +8,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {} class GridImagePool: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("IMAGE", "MASK", "INT", "INT", "STRING") RETURN_NAMES = ("image", "mask", "index", "count", "label") diff --git a/gates/profile_node.py b/gates/profile_node.py index 4d743aa..59ef3e2 100644 --- a/gates/profile_node.py +++ b/gates/profile_node.py @@ -4,7 +4,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {} class PoolProfile: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("POOL_PROFILE",) RETURN_NAMES = ("profile",) diff --git a/gates/sidecar_node.py b/gates/sidecar_node.py index 61577c4..179089c 100644 --- a/gates/sidecar_node.py +++ b/gates/sidecar_node.py @@ -11,7 +11,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {} class Sidecar: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("SIDECAR",) RETURN_NAMES = ("sidecar",) @@ -34,7 +34,7 @@ class Sidecar: class SaveImageSidecars: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "save" RETURN_TYPES = () OUTPUT_NODE = True diff --git a/gates/textgate.py b/gates/textgate.py index ece6b55..61eb758 100644 --- a/gates/textgate.py +++ b/gates/textgate.py @@ -15,7 +15,7 @@ ANY = AnyType("*") class TextGate: - CATEGORY = "Datasete Gates" + CATEGORY = "Dataset Gates" FUNCTION = "run" RETURN_TYPES = ("STRING", ANY) RETURN_NAMES = ("text", "signal")