Fix node def registeration (#803)

Comfy-Org · Sep 12, 2024 · 8889c4d · 8889c4d
1 parent 637f5b5
commit 8889c4d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/scripts/app.ts b/src/scripts/app.ts
@@ -2013,7 +2013,7 @@ export class ComfyApp {
       // TODO: change to "title?" once litegraph.d.ts has been updated
       static title = nodeData.display_name || nodeData.name
       static nodeData? = nodeData
-      static category: string = nodeData.category
+      static category?: string
 
       constructor(title?: string) {
         super(title)
@@ -2092,6 +2092,9 @@ export class ComfyApp {
 
     await this.#invokeExtensionsAsync('beforeRegisterNodeDef', node, nodeData)
     LiteGraph.registerNodeType(nodeId, node)
+    // Note: Do not move this to the class definition, it will be overwritten
+    // @ts-expect-error
+    node.category = nodeData.category
   }
 
   async registerNodesFromDefs(defs: Record<string, ComfyNodeDef>) {

diff --git a/src/types/comfyLGraphNode.ts b/src/types/comfyLGraphNode.ts
@@ -5,7 +5,7 @@ export declare class ComfyLGraphNode extends LGraphNode {
   static comfyClass: string
   static title: string
   static nodeData?: ComfyNodeDef
-  static category: string
+  static category?: string
 
   constructor(title?: string)
 }