chore: refurish python code by applying Pylint linter rules (#8322)

2024-09-13 22:42:08 +08:00 · 2024-09-13 22:42:08 +08:00 · a1104ab97e
commit a1104ab97e
parent 1ab81b4972
126 changed files with 253 additions and 272 deletions
--- a/api/core/workflow/nodes/end/end_stream_generate_router.py
+++ b/api/core/workflow/nodes/end/end_stream_generate_router.py
@ -136,10 +136,10 @@ class EndStreamGeneratorRouter:
        for edge in reverse_edges:
            source_node_id = edge.source_node_id
            source_node_type = node_id_config_mapping[source_node_id].get("data", {}).get("type")
-            if source_node_type in (
+            if source_node_type in {
                NodeType.IF_ELSE.value,
                NodeType.QUESTION_CLASSIFIER,
-            ):
+            }:
                end_dependencies[end_node_id].append(source_node_id)
            else:
                cls._recursive_fetch_end_dependencies(