626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Minor cleanup.

Browse source 
PiperOrigin-RevId: 392982022
This commit is contained in:
Galen Andrew 2021-08-25 14:34:23 -07:00 committed by A. Unique TensorFlower
parent 433b66b316
commit 9b48c81b6a
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tensorflow_privacy/privacy/analysis/dp_event_builder.py
View file

@ -62,14 +62,14 @@ class DpEventBuilder(object):
"""Builds and returns the composed DpEvent represented by the builder.""" """Builds and returns the composed DpEvent represented by the builder."""
if not self._composed_event: if not self._composed_event:
self_composed_events = [] self_composed_events = []
for event, num_self_compositions in self._events.items(): for event, count in self._events.items():
if num_self_compositions == 1: if count == 1:
self_composed_events.append(event) self_composed_events.append(event)
else: else:
self_composed_events.append( self_composed_events.append(
dp_event.SelfComposedDpEvent(event, num_self_compositions)) dp_event.SelfComposedDpEvent(event, count))
if not self_composed_events: if not self_composed_events:
return dp_event.NoOpDpEvent() self._composed_event = dp_event.NoOpDpEvent()
elif len(self_composed_events) == 1: elif len(self_composed_events) == 1:
self._composed_event = self_composed_events[0] self._composed_event = self_composed_events[0]
else: else: