Fix NODE_DEFAULT_SHAPE (#389)

* Fix NODE_DEFAULT_SHAPE * update snapshot
Comfy-Org · Dec 24, 2024 · 11ce8e0 · 11ce8e0
1 parent c8796f8
commit 11ce8e0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/LGraphCanvas.ts b/src/LGraphCanvas.ts
@@ -5145,7 +5145,7 @@ export class LGraphCanvas {
     const low_quality = this.ds.scale < 0.5
 
     const { collapsed } = node.flags
-    const shape = node._shape || node.constructor.shape || RenderShape.ROUND
+    const shape = node._shape || node.constructor.shape || LiteGraph.NODE_DEFAULT_SHAPE
     const { title_mode } = node.constructor
 
     const render_title = title_mode == TitleMode.TRANSPARENT_TITLE || title_mode == TitleMode.NO_TITLE

diff --git a/src/LiteGraphGlobal.ts b/src/LiteGraphGlobal.ts
@@ -51,7 +51,7 @@ export class LiteGraphGlobal {
   NODE_DEFAULT_COLOR = "#333"
   NODE_DEFAULT_BGCOLOR = "#353535"
   NODE_DEFAULT_BOXCOLOR = "#666"
-  NODE_DEFAULT_SHAPE = "box"
+  NODE_DEFAULT_SHAPE = RenderShape.ROUND
   NODE_BOX_OUTLINE_COLOR = "#FFF"
   NODE_ERROR_COLOUR = "#E00"
   DEFAULT_SHADOW_COLOR = "rgba(0,0,0,0.5)"

diff --git a/test/__snapshots__/litegraph.test.ts.snap b/test/__snapshots__/litegraph.test.ts.snap
@@ -54,7 +54,7 @@ LiteGraphGlobal {
   "NODE_DEFAULT_BGCOLOR": "#353535",
   "NODE_DEFAULT_BOXCOLOR": "#666",
   "NODE_DEFAULT_COLOR": "#333",
-  "NODE_DEFAULT_SHAPE": "box",
+  "NODE_DEFAULT_SHAPE": 2,
   "NODE_ERROR_COLOUR": "#E00",
   "NODE_MIN_WIDTH": 50,
   "NODE_MODES": [