feat(pipeline): add task configuration to RunConfig object

entelecheia · Jun 26, 2023 · 9c16b6a · 9c16b6a
1 parent 8c4a595
commit 9c16b6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/hyfi/pipeline/configs.py b/src/hyfi/pipeline/configs.py
@@ -7,6 +7,7 @@
 
 from hyfi.composer import Composer
 from hyfi.composer.extended import XC
+from hyfi.task import TaskConfig
 from hyfi.utils.envs import ENVs
 from hyfi.utils.logging import LOGGING
 
@@ -57,6 +58,7 @@ class RunConfig(BaseRunConfig):
 
     _target_: str = ""
     pipe_obj_arg_name: Optional[str] = ""
+    task: Optional[TaskConfig] = None
 
     def set_enviroment(self):
         if self.env: