def __init__()

in privacy_lint/attacks/grad_norm.py [0:0]

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

    def __init__(
        self,
        compute_grad_norm: Callable[
            [nn.Module, DataLoader], torch.Tensor
        ] = compute_grad_norm_cross_entropy,