Remove unnecessary string

OpenMined · Jan 11, 2022 · efcb92c · efcb92c
1 parent ceabdbd
commit efcb92c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pipeline_dp/private_beam.py b/pipeline_dp/private_beam.py
@@ -79,7 +79,7 @@ def __init__(self,
         self._sum_params = sum_params
 
     def expand(self, pcol: pvalue.PCollection) -> pvalue.PCollection:
-        beam_operations = pipeline_dp.BeamOperations("Expand")
+        beam_operations = pipeline_dp.BeamOperations()
         dp_engine = pipeline_dp.DPEngine(self._budget_accountant,
                                          beam_operations)
 

diff --git a/tests/dp_engine_test.py b/tests/dp_engine_test.py
@@ -529,7 +529,7 @@ def test_run_e2e_beam(self):
             input = p | "Create input" >> beam.Create(list(range(10)))
 
             output = self.run_e2e_private_partition_selection_large_budget(
-                input, pipeline_dp.BeamOperations("e2e_test"))
+                input, pipeline_dp.BeamOperations())
 
             beam_util.assert_that(output, beam_util.is_not_empty())