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()}