brainrot mistake

RomiconEZ · Jan 25, 2025 · 46dceff · 46dceff
1 parent 515d83f
commit 46dceff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llamator/attacks/bon.py b/src/llamator/attacks/bon.py
@@ -176,6 +176,7 @@ def run(self) -> Generator[StatusUpdate, None, None]:
             self.num_attempts * (self.multistage_depth + 1),
         )
 
+        total_attack_num = len(attack_prompts)
         for attack_prompt_index, attack_prompt in enumerate(attack_prompts):
             chat = ChatSession(
                 self.client_config.get_target_client(),
@@ -204,7 +205,6 @@ def run(self) -> Generator[StatusUpdate, None, None]:
             responses.append(response)
             statuses.append(current_status)
 
-            total_attack_num = len(attack_prompts)
             yield StatusUpdate(
                 self.client_config,
                 self.test_name,