FederatedAI · sagewe · Jul 19, 2022 · Jul 21, 2022 · Jul 21, 2022 · Jul 21, 2022
diff --git a/python/fate_arch/tensor/__init__.py b/python/fate_arch/tensor/__init__.py
@@ -1,19 +1,20 @@
 from ._dataloader import LabeledDataloaderWrapper, UnlabeledDataloaderWrapper
-from ._federation import ARBITER, GUEST, HOST
-from ._context import Context, CipherKind
-from ._tensor import (
-    FPTensor,
-    PHETensor,
-)
+from ._parties import Parties, PreludeParty
+from ._tensor import CipherKind, Context, FPTensor, PHETensor
+
+ARBITER = PreludeParty.ARBITER
+GUEST = PreludeParty.GUEST
+HOST = PreludeParty.HOST
 
 __all__ = [
     "FPTensor",
     "PHETensor",
+    "Parties",
     "ARBITER",
     "GUEST",
     "HOST",
     "Context",
     "LabeledDataloaderWrapper",
     "UnlabeledDataloaderWrapper",
-    "CipherKind"
+    "CipherKind",
 ]
diff --git a/python/fate_arch/tensor/_context.py b/python/fate_arch/tensor/_context.py
diff --git a/python/fate_arch/tensor/_federation.py b/python/fate_arch/tensor/_federation.py