update

util/injection.py

@@ -60,7 +60,7 @@ def process_scores_multiple(df, num_run,parameters,privilege_label,protect_label
 
     for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
         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]):
+            for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
                 prompt_temp = create_summary(row,group_name,label,occupation)
                 print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
                 print("=============================================================")
@@ -82,7 +82,7 @@ def process_scores_single(df, num_run,parameters,counterfactual_label,agent,grou
 
     for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
         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]):
+            for key, label in zip(['Counterfactual', 'Neutral'], [counterfactual_label, False]):
                 prompt_temp = create_summary(row,group_name,label,occupation)
                 print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
                 print("=============================================================")