From 61c79b0013ad3ac8fe8e825a225b6fabd08fb0ef Mon Sep 17 00:00:00 2001 From: -LAN- Date: Thu, 4 Sep 2025 02:15:46 +0800 Subject: [PATCH] test: correct imported name --- .../core/workflow/graph_engine/test_graph_engine.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py b/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py index f0774f7a29..11eecb6d77 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py +++ b/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py @@ -498,10 +498,10 @@ def test_layer_system_basic(): def test_layer_chaining(): """Test chaining multiple layers.""" - from core.workflow.graph_engine.layers import DebugLoggingLayer, Layer + from core.workflow.graph_engine.layers import DebugLoggingLayer, GraphEngineLayer # Create a custom test layer - class TestLayer(Layer): + class TestLayer(GraphEngineLayer): def __init__(self): super().__init__() self.events_received = [] @@ -560,10 +560,10 @@ def test_layer_chaining(): def test_layer_error_handling(): """Test that layer errors don't crash the engine.""" - from core.workflow.graph_engine.layers import Layer + from core.workflow.graph_engine.layers import GraphEngineLayer # Create a layer that throws errors - class FaultyLayer(Layer): + class FaultyLayer(GraphEngineLayer): def on_graph_start(self): raise RuntimeError("Intentional error in on_graph_start")