def additive_noise()

in augment/effects.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def additive_noise(self, noise_generator: Callable, snr: float):
        """
        >>> signal = torch.zeros((1, 100)).uniform_()
        >>> noise_generator = lambda: torch.zeros((1, 100))
        >>> chain = EffectChain().additive_noise(noise_generator, snr=0)
        >>> x = chain.apply(signal, {'rate': 16000}, {'rate': 16000})
        >>> (x == signal.mul(0.5)).all().item()
        True
        """
        self._chain.append(AdditiveNoise(
            noise_generator=noise_generator, snr=snr))
        return self