update

util/injection.py

@@ -63,8 +63,8 @@ def process_scores_multiple(df, num_run,parameters,privilege_label,protect_label
         for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, None]):
                 prompt_temp = create_summary(row,group_name,label,occupation)
-                # print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
-                # print("=============================================================")
+                print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
+                print("=============================================================")
                 result = invoke_retry(prompt_temp,agent,parameters)
                 scores[key][index].append(result)
 
@@ -85,8 +85,8 @@ def process_scores_single(df, num_run,parameters,counterfactual_label,agent,grou
         for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
             for key, label in zip(['Counterfactual', 'Neutral'], [counterfactual_label, None]):
                 prompt_temp = create_summary(row,group_name,label,occupation)
-                # print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
-                # print("=============================================================")
+                print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
+                print("=============================================================")
                 result = invoke_retry(prompt_temp,agent,parameters)
                 scores[key][index].append(result)