|
17 | 17 |
|
18 | 18 | from __future__ import annotations |
19 | 19 |
|
20 | | -__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY", "comfy_entrypoint"] |
| 20 | +__all__ = ["WEB_DIRECTORY", "comfy_entrypoint"] |
21 | 21 |
|
22 | 22 | from comfy_api.latest import io, ComfyExtension |
23 | 23 |
|
|
31 | 31 | WEB_DIRECTORY = "./web" |
32 | 32 |
|
33 | 33 |
|
34 | | -# Generate NODE_CLASS_MAPPINGS and NODE_DISPLAY_NAME_MAPPINGS from V3 nodes |
35 | | -# This is required for the ComfyUI Registry to discover nodes via static analysis |
36 | | -def _generate_node_mappings( |
37 | | - nodes: list[type[io.ComfyNode]], |
38 | | -) -> tuple[dict[str, type[io.ComfyNode]], dict[str, str]]: |
39 | | - """Generate legacy NODE_CLASS_MAPPINGS from V3 node classes.""" |
40 | | - class_mappings = {} |
41 | | - display_name_mappings = {} |
42 | | - |
43 | | - for node_class in nodes: |
44 | | - schema = node_class.define_schema() |
45 | | - node_id = schema.node_id |
46 | | - display_name = schema.display_name or node_id |
47 | | - |
48 | | - class_mappings[node_id] = node_class |
49 | | - display_name_mappings[node_id] = display_name |
50 | | - |
51 | | - return class_mappings, display_name_mappings |
52 | | - |
53 | | - |
54 | | -NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = _generate_node_mappings(RTX_REMIX_NODES) |
55 | | - |
56 | | - |
57 | 34 | class RTXRemixExtension(ComfyExtension): |
58 | 35 | """RTX Remix Extension for ComfyUI""" |
59 | 36 |
|
|
0 commit comments