update

util/injection.py

@@ -71,7 +71,6 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
     print(f"Processing {len(df)} entries with {num_run} runs each.")
     """ Process entries and compute scores concurrently, with progress updates. """
     scores = {key: [[] for _ in range(len(df))] for key in ['Privilege','Protect','Neutral']}
-    prompt_list = []
 
     for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
         for index, (idx, row) in tqdm(enumerate(df.iterrows()), total=len(df), desc="Processing entries", unit="entry"):
@@ -85,7 +84,6 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
                 print("=============================================================")
                 result_normal = invoke_retry(prompt_normal, agent, parameters)
                 scores[key][index].append(result_normal)
-                prompt_list.append(prompt_normal)
 
     print(f"Scores: {scores}")
 
@@ -109,6 +107,4 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
     df['Protect_Rank'] = ranks['Protect_Avg_Score']
     df['Neutral_Rank'] = ranks['Neutral_Avg_Score']
 
-    df['prompt'] = prompt_list
-
     return df