[PoB] Fix BoundedSumFloat64 being used in instead of BoundedSumInt64

PiperOrigin-RevId: 608613104 Change-Id: Id3a9a50ce057352c52c51e90eff71d8f43b8d84a GitOrigin-RevId: ba65bdca42a1722d05f1fb07c8b62f4e8ff22c82
google · Feb 21, 2024 · dc6456b · dc6456b
1 parent a9c0c38
commit dc6456b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/privacy-on-beam/pbeam/aggregations_test.go b/privacy-on-beam/pbeam/aggregations_test.go
@@ -277,7 +277,7 @@ func TestBoundedSumInt64FnExtractOutputReturnsNilForSmallPartitions(t *testing.T
 		{"Input with 1 privacy unit", 1},
 	} {
 		spec := privacySpec(t, PrivacySpecParams{AggregationEpsilon: 0.5, PartitionSelectionEpsilon: 0.5, PartitionSelectionDelta: 1e-23})
-		fn, err := newBoundedSumFloat64Fn(
+		fn, err := newBoundedSumInt64Fn(
 			*spec,
 			SumParams{AggregationEpsilon: 0.5, PartitionSelectionParams: PartitionSelectionParams{Epsilon: 0.5, Delta: 1e-23}, MaxPartitionsContributed: 1, MinValue: 0, MaxValue: 2},
 			noise.LaplaceNoise,