fix engine test bug caused by no-clear between tests (#350)

LazyAGI · Nov 20, 2024 · 3c1d3a0 · 3c1d3a0
1 parent 8035a65
commit 3c1d3a0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lazyllm/engine/engine.py b/lazyllm/engine/engine.py
@@ -465,7 +465,7 @@ def cond(x):
                     return True
         return False
 
-    return lazyllm.ifs(cond, tpath=lazyllm.TrainableModule(base_model), fpath=lazyllm.Identity())
+    return lazyllm.ActionModule(lazyllm.ifs(cond, tpath=lazyllm.TrainableModule(base_model), fpath=lazyllm.Identity()))
 
 
 @NodeConstructor.register('Constant')

diff --git a/tests/advanced_tests/standard_test/test_engine.py b/tests/advanced_tests/standard_test/test_engine.py
@@ -1,5 +1,6 @@
 import lazyllm
 import os
+import pytest
 from lazyllm.engine import LightEngine
 
 class TestEngine(object):
@@ -11,6 +12,13 @@ def _test_vqa(self):
         engine = LightEngine()
         engine.start(node, edge, resource)
 
+    @pytest.fixture(autouse=True)
+    def run_around_tests(self):
+        yield
+        LightEngine().reset()
+        lazyllm.FileSystemQueue().dequeue()
+        lazyllm.FileSystemQueue(klass="lazy_trace").dequeue()
+
     def test_http(self):
         nodes = [
             dict(