fix(configs): add use_pipe_obj and return_pipe_obj flags

entelecheia · Jun 26, 2023 · 6b10ceb · 6b10ceb
1 parent 3d217a0
commit 6b10ceb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/hyfi/pipeline/configs.py b/src/hyfi/pipeline/configs.py
@@ -59,7 +59,9 @@ class PipeConfig(BaseRunConfig):
 
     _pipe_: str = ""
     _run_: str = ""
+    use_pipe_obj: bool = True
     pipe_obj_arg_name: Optional[str] = ""
+    return_pipe_obj: bool = False
     task: Optional[TaskConfig] = None
 
     def set_enviroment(self):