From e7dfaa4eae9e52f6f5655029941d86e179311b47 Mon Sep 17 00:00:00 2001 From: sagewe Date: Tue, 22 Aug 2023 14:45:34 +0800 Subject: [PATCH] improve decrypt Signed-off-by: sagewe --- python/fate/arch/histogram/histogram.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/fate/arch/histogram/histogram.py b/python/fate/arch/histogram/histogram.py index da92926476..ed82df0bf9 100644 --- a/python/fate/arch/histogram/histogram.py +++ b/python/fate/arch/histogram/histogram.py @@ -478,7 +478,7 @@ def decrypt( sk_map: MutableMapping[str, typing.Any], coder_map: MutableMapping[str, typing.Tuple[typing.Any, torch.dtype]], ): - out = list(self._table.map(lambda pid, split: (pid, split.decrypt(sk_map, coder_map))).collect()) + out = list(self._table.mapValues(lambda split: split.decrypt(sk_map, coder_map)).collect()) out.sort(key=lambda x: x[0]) return self.cat([split for _, split in out])