def step()

in opacus/accountants/gdp.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

    def step(self, *, noise_multiplier: float, sample_rate: float):
        if self.noise_multiplier is None:
            self.noise_multiplier = noise_multiplier

        if self.sample_rate is None:
            self.sample_rate = sample_rate

        if noise_multiplier != self.noise_multiplier or sample_rate != self.sample_rate:
            raise ValueError(
                "Noise multiplier and sample rate have to stay constant in GaussianAccountant."
            )
        self.steps += 1