fix cipher

Signed-off-by: weiwee <wbwmat@gmail.com>
FederatedAI · Jul 11, 2023 · 4b66762 · 4b66762
1 parent 05fd550
commit 4b66762
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fate/arch/context/_cipher.py b/python/fate/arch/context/_cipher.py
@@ -50,4 +50,4 @@ def __init__(self, kind) -> None:
     def keygen(self, **kwargs):
         from fate.arch.tensor import phe_keygen
 
-        return phe_keygen(self.device, **kwargs)
+        return phe_keygen(self.kind, **kwargs)