Step 1: save plan, clean up old-style node registration and broken tests

This commit is contained in:
Immac
2026-03-01 16:38:08 -06:00
parent 3e39e9bb09
commit 4fb0aaa4bb
4 changed files with 61 additions and 63 deletions
+2 -38
View File
@@ -1,40 +1,4 @@
"""ComfyUI-ImmacTools package exports used by ComfyUI.
"""ComfyUI-ImmacTools node package.
This module exposes the node registration mappings and the `WEB_DIRECTORY`
constant expected by ComfyUI web extensions.
Node registration is handled via comfy_entrypoint() in nodes.py.
"""
from .nodes import ConcatenateSigmasNode, SpliceSigmasAtNode, ResampleSigmas, SkipEveryNthImages, MatchContrastNode
from .forwarding_nodes import ForwardAnyNode, ForwardConditioningNode, ForwardModelNode
# Mapping from node_id to node class
NODE_CLASS_MAPPINGS: dict[str, type] = {
"ConcatenateSigmasImmacTools": ConcatenateSigmasNode,
"SpliceSigmasAtImmacTools": SpliceSigmasAtNode,
"ResampleSigmasImmacTools": ResampleSigmas,
"SkipEveryNthImagesImmacTools": SkipEveryNthImages,
"MatchContrastImmacTools": MatchContrastNode,
"ForwardAnyImmacTools": ForwardAnyNode,
"ForwardConditioningImmacTools": ForwardConditioningNode,
"ForwardModelImmacTools": ForwardModelNode,
}
# Mapping from node_id to human readable display name
NODE_DISPLAY_NAME_MAPPINGS: dict[str, str] = {
"ConcatenateSigmasImmacTools": "Concatenate Sigmas Node",
"SpliceSigmasAtImmacTools": "Splice Sigmas At Node",
"ResampleSigmasImmacTools": "Resample Sigmas",
"SkipEveryNthImagesImmacTools": "Skip Every Nth Image",
"MatchContrastImmacTools": "Match Contrast",
"ForwardAnyImmacTools": "Forward Any",
"ForwardConditioningImmacTools": "Forward Conditioning",
"ForwardModelImmacTools": "Forward Model",
}
# ComfyUI expects this for web extensions
WEB_DIRECTORY = "./web"
__all__ = [
"NODE_CLASS_MAPPINGS",
"NODE_DISPLAY_NAME_MAPPINGS",
"WEB_DIRECTORY",
]
+1 -4
View File
@@ -109,10 +109,7 @@ class ForwardModelNode(io.ComfyNode):
return io.NodeOutput(model)
class ExampleForwardingExtension(ComfyExtension):
@override
async def get_node_list(self) -> list[type[io.ComfyNode]]:
return [ForwardAnyNode, ForwardConditioningNode, ForwardModelNode]