Merge branch 'dev-2.0.0-beta-debugging' of https://github.com/Federat…

…edAI/FATE into dev-2.0.0-beta-debugging

python/fate/arch/tensor/distributed/_tensor.py

@@ -81,6 +81,9 @@ def __matmul__(self, other):
     def __rmatmul__(self, other):
         return torch.matmul(other, self)
 
+    def encrypt(self, encryptor):
+        return torch.encrypt_f(self, encryptor)
+
     @property
     def shape(self):
         return self.shardings.shape

python/fate/components/__main__.py

@@ -15,7 +15,7 @@
 #
 
 """
-execute with python -m fate.components --execution_id xxx --config xxx
+execute with python -m fate.components --config xxx
 """
 
 if __name__ == "__main__":

python/fate/components/core/params/_optimizer.py

@@ -1,16 +1,20 @@
+from typing import Type
+
 import pydantic
 
 from ._fields import string_choice
 from ._penalty import penalty_param
 
 
 class OptimizerParam(pydantic.BaseModel):
-    method: string_choice(['sgd', 'adadelta', 'adagrad', 'adam', 'adamax', 'adamw',
-                           'asgd', 'nadam', 'radam', 'rmsprop', 'rprop']) = 'sgd'
-    penalty: penalty_param(l1=True, l2=True, none=True) = 'l2'
+    method: string_choice(
+        ["sgd", "adadelta", "adagrad", "adam", "adamax", "adamw", "asgd", "nadam", "radam", "rmsprop", "rprop"]
+    ) = "sgd"
+    penalty: penalty_param(l1=True, l2=True, none=True) = "l2"
     alpha: float = 1.0
     optimizer_params: dict
 
-def optimizer_param():
+
+def optimizer_param() -> Type[OptimizerParam]:
     namespace = {}
     return type("OptimizerParam", (OptimizerParam,), namespace)