Merge pull request #5065 from FederatedAI/dev-2.0.0-beta-improve-iupdate

dev 2.0.0 beta improve iupdate
FederatedAI · Aug 22, 2023 · e30299c · e30299c
2 parents 09e25e5 + e7dfaa4
commit e30299c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])