From bbb8f35f49adc771193fea36214f4db7fecda9f8 Mon Sep 17 00:00:00 2001
From: Great <44393369+yaosheng216@users.noreply.github.com>
Date: Fri, 14 Feb 2025 16:43:11 +0800
Subject: [PATCH] Update __init__.py

---
 scepter/workflow/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scepter/workflow/__init__.py b/scepter/workflow/__init__.py
index 7ddba34..c5b4387 100644
--- a/scepter/workflow/__init__.py
+++ b/scepter/workflow/__init__.py
@@ -7,6 +7,7 @@ from .mantras_node import MantrasNode
 from .tuner_node import TunerNode
 from .control_node import ControlNode
 from .ace_plus_processor_node import ACEPlusProcessorNode
+from .calculator_node import CalculatorNode
 
 
 NODE_MAPPINGS = {
@@ -16,7 +17,8 @@ NODE_MAPPINGS = {
     'MantrasNode': ('🪄 ScepterMantra~', MantrasNode),
     'TunerNode': ('🪄 ScepterTuner~', TunerNode),
     'ControlNode': ('🪄 ScepterControl~', ControlNode),
-    'ACEPlusProcessorNode': ('🪄 ScepterACEPlusImageProcessor~', ACEPlusProcessorNode)
+    'ACEPlusProcessorNode': ('🪄 ScepterACEPlusImageProcessor~', ACEPlusProcessorNode),
+    'CalculatorNode': ('🪄 ScepterCalculator~', CalculatorNode)
 }
 
 NODE_CLASS_MAPPINGS = {k: v[1] for k, v in NODE_MAPPINGS.items()}