Merge pull request #5381 from FederatedAI/feature-2.0.0-rc-remove-ml-mpc

chore: remove unused code
FederatedAI · Dec 27, 2023 · e18ef6b · e18ef6b
2 parents c131f49 + 9fbca1a
commit e18ef6b
Show file tree

Hide file tree

Showing 10 changed files with 3 additions and 1,061 deletions.
diff --git a/python/fate/components/core/_cpn_task_mode.py b/python/fate/components/core/_cpn_task_mode.py
@@ -1,15 +1,17 @@
 import enum
 import os
 
+
 class TaskMode(enum.Enum):
     SIMPLE = "SIMPLE"
     DEEPSPEED = "DEEPSPEED"
 
     def __str__(self):
         return self.value
 
+
 def is_deepspeed_mode():
-    return os.getenv("FATE_TASK_TYPE", "").upper() == TaskMode.DEEPSPEED
+    return os.getenv("FATE_TASK_TYPE", "").upper() == TaskMode.DEEPSPEED.value
 
 
 def is_root_worker():

diff --git a/python/fate/ml/mpc/__init__.py b/python/fate/ml/mpc/__init__.py