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
This commit is contained in:
Galen Andrew 2021-09-22 11:30:33 -07:00 committed by A. Unique TensorFlower
parent c39d628e16
commit 67a7096d52

View file

@ -74,7 +74,7 @@ class SelfComposedDpEvent(DpEvent):
@attr.s(frozen=True, slots=True, auto_attribs=True) @attr.s(frozen=True, slots=True, auto_attribs=True)
class ComposedDpEvent(DpEvent): class ComposedDpEvent(DpEvent):
"""A series of composed mechanisms.""" """A series of composed mechanisms."""
events: List[SelfComposedDpEvent] events: List[DpEvent]
@attr.s(frozen=True, slots=True, auto_attribs=True) @attr.s(frozen=True, slots=True, auto_attribs=True)