feat: enable prefect for CLI

FAST-HEP · Mar 19, 2024 · dbeb763 · dbeb763
1 parent f1b5751
commit dbeb763
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/fasthep_flow/cli.py b/src/fasthep_flow/cli.py
@@ -7,6 +7,7 @@
 import typer
 
 from .config import FlowConfig, load_config
+from .orchestration import prefect_workflow
 from .workflow import Workflow
 
 app = typer.Typer()
@@ -51,7 +52,9 @@ def execute(
     # initialize hydra
     cfg = init_config(config, overrides)
     workflow = Workflow(config=cfg)
-    results = workflow.run()
+    flow = prefect_workflow(workflow)
+    results = flow()
+    # results = workflow.run()
     typer.echo(f"Results: {results}")