Removed node_replace files, moved NodeReplace to _io.py and exposed it

This commit is contained in:
Jedrzej Kosinski
2026-02-14 20:34:25 -08:00
parent a9cc84bb78
commit 2d992ebd6c
7 changed files with 75 additions and 85 deletions

View File

@@ -10,7 +10,6 @@ from ._input_impl import VideoFromFile, VideoFromComponents
from ._util import VideoCodec, VideoContainer, VideoComponents, MESH, VOXEL, File3D
from . import _io_public as io
from . import _ui_public as ui
from . import _node_replace_public as node_replace
from comfy_execution.utils import get_executing_context
from comfy_execution.progress import get_progress_state, PreviewImageTuple
from PIL import Image
@@ -28,7 +27,7 @@ class ComfyAPI_latest(ComfyAPIBase):
self.execution = self.Execution()
class NodeReplacement(ProxiedSingleton):
async def register(self, node_replace: 'node_replace.NodeReplace') -> None:
async def register(self, node_replace: io.NodeReplace) -> None:
"""Register a node replacement mapping."""
from server import PromptServer
PromptServer.instance.node_replace_manager.register(node_replace)
@@ -141,5 +140,4 @@ __all__ = [
"IO",
"ui",
"UI",
"node_replace",
]