From 7507b081997eaba9b2704e86143de4562f06c93c Mon Sep 17 00:00:00 2001 From: jodi spacek Date: Wed, 22 Dec 2021 16:56:49 -0500 Subject: [PATCH] Fix private beam --- pipeline_dp/private_beam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipeline_dp/private_beam.py b/pipeline_dp/private_beam.py index 593b060d..1579c2db 100644 --- 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)