fix progress bars for adversarial

Author: dylanbouchard

langfair/generator/generator.py

@@ -216,9 +216,9 @@ async def generate_responses(
             assert count == 1, "temperature must be greater than 0 if count > 1"
         self._update_count(count)
         self.system_message = SystemMessage(system_prompt)
-
+                
+        total = len(prompts) * self.count
         if show_progress_bars:
-            total = len(prompts) * self.count
             self.progress_bar = start_progress_bar(existing_progress_bar)
             self.progress_task = self.progress_bar.add_task(
                 f"Generating {self.count} responses per prompt...",

langfair/generator/redteaming.py

@@ -224,7 +224,6 @@ async def _generate_from_template(
             )
         dataset = {}
         for system_style in system_styles:
-            print(f"Generating responses with {system_style} system prompts...")
             system_prompt = SYSTEM_PROMPT_DICT[system_style]
             with contextlib.redirect_stdout(io.StringIO()):
                 tmp = await self.generate_responses(