Merge pull request #5345 from FederatedAI/feature-2.0.0-rc-component-…

…blacklist add components blacklist
FederatedAI · Dec 19, 2023 · 4c2337b · 4c2337b
2 parents b5ff7cd + 4948b7e
commit 4c2337b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/configs/default.yaml b/configs/default.yaml
@@ -1,3 +1,6 @@
+components:
+  blacklist: []
+
 safety:
   serdes:
     # supported types: unrestricted, restricted

diff --git a/python/fate/arch/config/_config.py b/python/fate/arch/config/_config.py
@@ -63,3 +63,7 @@ def federation(self):
     def safety(self):
         return self.config.safety
 
+    @property
+    def components(self):
+        return self.config.components
+
diff --git a/python/fate/components/entrypoint/cli/component/execute_cli.py b/python/fate/components/entrypoint/cli/component/execute_cli.py
@@ -79,6 +79,12 @@ def execute(
     logger.debug(f"task config: {task_config}")
 
     os.makedirs(os.path.dirname(execution_final_meta_path), exist_ok=True)
+
+    from fate.arch.config import cfg
+
+    if task_config.component in cfg.components.blacklist:
+        raise RuntimeError(f"component `{task_config.component}` is in blacklist, do not use it")
+
     execute_component_from_config(task_config, execution_final_meta_path)