Merge pull request #5174 from FederatedAI/dev-2.0.0-beta-fix-zeros-bug

fix zeros bug
FederatedAI · Sep 12, 2023 · 030a4d2 · 030a4d2
2 parents 1944490 + 65c80ed
commit 030a4d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fate/arch/histogram/values/_cipher.py b/python/fate/arch/histogram/values/_cipher.py
@@ -85,7 +85,7 @@ def compute_child(
         positions: List[Tuple[int, int, int, int, int, int, int, int]],
         size: int,
     ):
-        data = self.evaluator.zeros(size * self.stride)
+        data = self.evaluator.zeros(size * self.stride, self.dtype)
         for (
             target_weak_child_start,
             target_weak_child_end,