Function get_attack_fn

agent_attack/attacks/__init__.py:6–14

(attack)

def get_attack_fn(attack):
    if attack == "pgd":
        return pgd
    elif attack == "bim":
        return bim
    elif attack == "clip_attack":
        return clip_attack
    else:
        raise ValueError(f"Attack {attack} not supported.")