feat(hydra-main): add plugins parameter

entelecheia · Jul 24, 2023 · 2b708ac · 2b708ac
1 parent 1a4c833
commit 2b708ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/hyfi/core/hydra/main.py b/src/hyfi/core/hydra/main.py
@@ -21,6 +21,7 @@ def main(
     config_name: Optional[str] = None,
     version_base: Optional[str] = _UNSPECIFIED_,
     overrides: Optional[List[str]] = None,
+    plugins: Optional[List[str]] = None,
 ) -> Callable[[TaskFunction], Any]:
     """
     :param config_path: The config path, a directory where Hydra will search for
@@ -75,6 +76,7 @@ def decorated_main(cfg_passthrough: Optional[DictConfig] = None) -> Any:
                     task_function=task_function,
                     config_path=config_path,
                     config_name=config_name,
+                    plugins=plugins,
                 )
 
         return decorated_main