diff --git a/garak/probes/tap.py b/garak/probes/tap.py index 1b811b44..10cbe32c 100644 --- a/garak/probes/tap.py +++ b/garak/probes/tap.py @@ -260,6 +260,7 @@ class PAIR(Probe): "depth": 10, "n_streams": 1, "keep_last_n": 1, + "pruning": True, } def __init__(self, config_root=_config): @@ -284,7 +285,7 @@ def probe(self, generator) -> List[garak.attempt.Attempt]: attack_model_config=self.attack_model_config, attack_max_attempts=self.attack_max_attempts, evaluator_model_type=self.evaluator_model_type, - evaluator_model=self.evaluator_model_name, + evaluator_model_name=self.evaluator_model_name, evaluator_model_config=self.evaluator_model_config, branching_factor=self.branching_factor, width=self.width,