def __init__(self, gain=1, distribution='normal', **kwargs):
    super().__init__(**kwargs)
    self.gain = gain
    self.distribution = distribution
def __init__(self, gain=1, distribution='normal', **kwargs):
    super().__init__(**kwargs)
    self.gain = gain
    self.distribution = distribution