Another fix

OpenMined · Jan 11, 2022 · 67cbab2 · 67cbab2
1 parent 990743c
commit 67cbab2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipeline_dp/pipeline_operations.py b/pipeline_dp/pipeline_operations.py
@@ -293,7 +293,7 @@ def sample_fixed_per_key(self, rdd, n: int, stage_name: str = None):
     def count_per_element(self, rdd, stage_name: str = None):
         return rdd.map(lambda x: (x, 1)).reduceByKey(lambda x, y: (x + y))
 
-    def reduce_accumulators_per_key(self, rdd, stage_name: str = None):
+    def reduce_accumulators_per_key(self, rdd, stage_name: str):
         return rdd.reduceByKey(lambda acc1, acc2: acc1.add_accumulator(acc2))
 
     def is_serialization_immediate_on_reduce_by_key(self):