(node_class: Type)
| 90 | """ |
| 91 | |
| 92 | def decorator(node_class: Type) -> Type: |
| 93 | if skip: |
| 94 | return node_class |
| 95 | |
| 96 | if not isinstance(node_class, type): |
| 97 | raise ValueError("`node_class` must be a class") |
| 98 | |
| 99 | nonlocal name, description |
| 100 | if name is None: |
| 101 | name = node_class.__name__ |
| 102 | |
| 103 | # Remove possible "Node" suffix from the class name, e.g. "EditImageNode -> EditImage" |
| 104 | if name is not None and name.endswith("Node"): |
| 105 | name = name[:-4] |
| 106 | |
| 107 | description = _format_description(description, name, experimental, deprecated) |
| 108 | |
| 109 | # For v3 nodes, wrap define_schema to inject the display_name |
| 110 | if _is_v3_node(node_class): |
| 111 | _wrap_define_schema(node_class, description) |
| 112 | |
| 113 | register_node(node_class, name, description) |
| 114 | return node_class |
| 115 | |
| 116 | # If the decorator is used without parentheses |
| 117 | if node_class is None: |
no test coverage detected