diff --git a/dagflow/init.py b/dagflow/init.py index ed1b2c8..b05bf4b 100644 --- a/dagflow/init.py +++ b/dagflow/init.py @@ -3,7 +3,7 @@ from functools import partial from appPublic.jsonConfig import getConfig from ahserver.serverenv import ServerEnv from ahserver.configuredServer import add_cleanupctx -from .dagflow import FlowEngine +from .dagflow import FlowEngine, add_new_workflow async def dagbacktask(engines, app): tasks = [] @@ -26,4 +26,5 @@ def load_dagflow(): add_cleanupctx(f) env = ServerEnv() env.flow_engines = flow_engines + env.add_new_workflow = add_new_workflow