Fix tap.PAIR run error - run_tap param evaluator_model (#1045)

NVIDIA · Dec 16, 2024 · 3f004ac · 3f004ac
2 parents 2714226 + d071c96
commit 3f004ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,