mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-15 14:09:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Abstracted out NodeOutput into _NodeOutputInternal in execution.py

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-07-24 16:58:25 -07:00
parent d3a62a440f
commit 44afeab124
3 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
comfy_api/internal/__init__.py
View File

@@ -36,6 +36,13 @@ class _ComfyNodeInternal:
...
class _NodeOutputInternal:
"""Class that all V3-based APIs inherit from for NodeOutput.
This is intended to only be referenced within execution.py, as it has to handle all V3 APIs going forward."""
...
def as_pruned_dict(dataclass_obj):
'''Return dict of dataclass object with pruned None values.'''
return prune_dict(asdict(dataclass_obj))

4
comfy_api/v3/_io.py
View File

@@ -22,7 +22,7 @@ from comfy.samplers import CFGGuider, Sampler
from comfy.sd import CLIP, VAE
from comfy.sd import StyleModel as StyleModel_
from comfy_api.input import VideoInput
from comfy_api.internal import (_ComfyNodeInternal, classproperty, copy_class, first_real_override, is_class,
from comfy_api.internal import (_ComfyNodeInternal, _NodeOutputInternal, classproperty, copy_class, first_real_override, is_class,
prune_dict, shallow_clone_class)
from comfy_api.v3._resources import Resources, ResourcesLocal
from comfy_execution.graph import ExecutionBlocker
@@ -1486,7 +1486,7 @@ class ComfyNode(_ComfyNodeBaseInternal):
return ComfyNode
class NodeOutput:
class NodeOutput(_NodeOutputInternal):
'''
Standardized output of a node; can pass in any number of args and/or a UIOutput into 'ui' kwarg.
'''