From bc4edbfc2bb5526a062248589c9c1f3aee623fe1 Mon Sep 17 00:00:00 2001 From: takatost Date: Mon, 4 Mar 2024 23:34:28 +0800 Subject: [PATCH] lint fix --- api/core/callback_handler/workflow_event_trigger_callback.py | 2 +- api/core/workflow/callbacks/base_callback.py | 2 +- api/core/workflow/nodes/base_node.py | 4 ++-- api/core/workflow/nodes/start/start_node.py | 2 +- api/core/workflow/workflow_engine_manager.py | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/api/core/callback_handler/workflow_event_trigger_callback.py b/api/core/callback_handler/workflow_event_trigger_callback.py index 2f81f27426..e1d2413534 100644 --- a/api/core/callback_handler/workflow_event_trigger_callback.py +++ b/api/core/callback_handler/workflow_event_trigger_callback.py @@ -1,6 +1,6 @@ from core.app.app_queue_manager import AppQueueManager, PublishFrom from core.workflow.callbacks.base_callback import BaseWorkflowCallback -from models.workflow import WorkflowRun, WorkflowNodeExecution +from models.workflow import WorkflowNodeExecution, WorkflowRun class WorkflowEventTriggerCallback(BaseWorkflowCallback): diff --git a/api/core/workflow/callbacks/base_callback.py b/api/core/workflow/callbacks/base_callback.py index a564af498c..76fe4d96d5 100644 --- a/api/core/workflow/callbacks/base_callback.py +++ b/api/core/workflow/callbacks/base_callback.py @@ -1,6 +1,6 @@ from abc import abstractmethod -from models.workflow import WorkflowRun, WorkflowNodeExecution +from models.workflow import WorkflowNodeExecution, WorkflowRun class BaseWorkflowCallback: diff --git a/api/core/workflow/nodes/base_node.py b/api/core/workflow/nodes/base_node.py index a95a232ae6..6f28a3f104 100644 --- a/api/core/workflow/nodes/base_node.py +++ b/api/core/workflow/nodes/base_node.py @@ -1,5 +1,5 @@ from abc import abstractmethod -from typing import Optional, Type +from typing import Optional from core.workflow.entities.base_node_data_entities import BaseNodeData from core.workflow.entities.node_entities import NodeType @@ -8,7 +8,7 @@ from core.workflow.entities.variable_pool import VariablePool class BaseNode: _node_type: NodeType - _node_data_cls: Type[BaseNodeData] + _node_data_cls: type[BaseNodeData] def __init__(self, config: dict) -> None: self._node_id = config.get("id") diff --git a/api/core/workflow/nodes/start/start_node.py b/api/core/workflow/nodes/start/start_node.py index 014a146c93..e218cced3d 100644 --- a/api/core/workflow/nodes/start/start_node.py +++ b/api/core/workflow/nodes/start/start_node.py @@ -1,4 +1,4 @@ -from typing import Type, Optional +from typing import Optional from core.workflow.entities.node_entities import NodeType from core.workflow.entities.variable_pool import VariablePool diff --git a/api/core/workflow/workflow_engine_manager.py b/api/core/workflow/workflow_engine_manager.py index afa4dbb321..3ad36fe1d2 100644 --- a/api/core/workflow/workflow_engine_manager.py +++ b/api/core/workflow/workflow_engine_manager.py @@ -19,7 +19,7 @@ from core.workflow.nodes.variable_assigner.variable_assigner_node import Variabl from extensions.ext_database import db from models.account import Account from models.model import App, EndUser -from models.workflow import Workflow, WorkflowRunTriggeredFrom, WorkflowRun, WorkflowRunStatus, CreatedByRole +from models.workflow import CreatedByRole, Workflow, WorkflowRun, WorkflowRunStatus, WorkflowRunTriggeredFrom node_classes = { NodeType.START: StartNode,