Fix private beam

OpenMined · Dec 22, 2021 · 7507b08 · 7507b08
1 parent a7bc6ac
commit 7507b08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipeline_dp/private_beam.py b/pipeline_dp/private_beam.py
@@ -83,7 +83,7 @@ def __init__(self,
         self._sum_params = sum_params
 
     def expand(self, pcol: pvalue.PCollection) -> pvalue.PCollection:
-        beam_operations = pipeline_dp.BeamOperations()
+        beam_operations = pipeline_dp.BeamOperations("Expand")
         dp_engine = pipeline_dp.DPEngine(self._budget_accountant,
                                          beam_operations)