From 75559bcbf90168ab4cf5f0b04881b0e4b01d6835 Mon Sep 17 00:00:00 2001 From: takatost Date: Mon, 4 Mar 2024 02:06:27 +0800 Subject: [PATCH] replace block type to node type --- api/core/workflow/entities/NodeEntities.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/api/core/workflow/entities/NodeEntities.py b/api/core/workflow/entities/NodeEntities.py index d72b000dfb..80471cc702 100644 --- a/api/core/workflow/entities/NodeEntities.py +++ b/api/core/workflow/entities/NodeEntities.py @@ -19,14 +19,14 @@ class NodeType(Enum): VARIABLE_ASSIGNER = 'variable-assigner' @classmethod - def value_of(cls, value: str) -> 'BlockType': + def value_of(cls, value: str) -> 'NodeType': """ - Get value of given block type. + Get value of given node type. - :param value: block type value - :return: block type + :param value: node type value + :return: node type """ - for block_type in cls: - if block_type.value == value: - return block_type - raise ValueError(f'invalid block type value {value}') + for node_type in cls: + if node_type.value == value: + return node_type + raise ValueError(f'invalid node type value {value}')