From 6751fe5b26a2f1e4018a51c5bc475b53808e62cc Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Sun, 21 Jun 2026 16:28:38 +0200 Subject: [PATCH] feat: register FolderImageLoader in node mappings Co-Authored-By: Claude Opus 4.8 --- __init__.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index 6ae87c4..5c5c5da 100644 --- a/__init__.py +++ b/__init__.py @@ -8,8 +8,14 @@ WEB_DIRECTORY = "./web" # setup — in that case the relative imports would raise. Guard on __package__ # so the test suite can import `gates.*` without dragging in aiohttp/comfy. if __package__: - from .gates.node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS + from .gates.node import NODE_CLASS_MAPPINGS as _POOL_NODES, \ + NODE_DISPLAY_NAME_MAPPINGS as _POOL_NAMES + from .gates.loader import NODE_CLASS_MAPPINGS as _LOADER_NODES, \ + NODE_DISPLAY_NAME_MAPPINGS as _LOADER_NAMES from .gates import routes # noqa: F401 (registers aiohttp routes on import) + + NODE_CLASS_MAPPINGS = {**_POOL_NODES, **_LOADER_NODES} + NODE_DISPLAY_NAME_MAPPINGS = {**_POOL_NAMES, **_LOADER_NAMES} else: # pragma: no cover - exercised only under pytest collection NODE_CLASS_MAPPINGS = {} NODE_DISPLAY_NAME_MAPPINGS = {}