afix engine test bug caused by no-clear between tests

LazyAGI · Nov 20, 2024 · 0e89d64 · 0e89d64
1 parent 6153aad
commit 0e89d64
Showing 1 changed file with 1 addition 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')