From 67a7096d529a8d43191defa5183670279537c919 Mon Sep 17 00:00:00 2001 From: Galen Andrew Date: Wed, 22 Sep 2021 11:30:33 -0700 Subject: [PATCH] ComposedDpEvent can be a list of any DpEvent, not only SelfComposedDpEvent. For example there is no reason we shouldn't be able to compose a single GaussianDpEvent and a single LaplaceDpEvent without having to wrap them in SelfComposedDpEvent with count == 1. PiperOrigin-RevId: 398288473 --- tensorflow_privacy/privacy/analysis/dp_event.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_privacy/privacy/analysis/dp_event.py b/tensorflow_privacy/privacy/analysis/dp_event.py index ac45a64..b40b823 100644 --- a/tensorflow_privacy/privacy/analysis/dp_event.py +++ b/tensorflow_privacy/privacy/analysis/dp_event.py @@ -74,7 +74,7 @@ class SelfComposedDpEvent(DpEvent): @attr.s(frozen=True, slots=True, auto_attribs=True) class ComposedDpEvent(DpEvent): """A series of composed mechanisms.""" - events: List[SelfComposedDpEvent] + events: List[DpEvent] @attr.s(frozen=True, slots=True, auto_attribs=True)