update

util/injection.py

@@ -84,6 +84,13 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
     raise Exception("Failed to complete the API call after maximum retry attempts.")
 
 
+def calculate_avg_score(score_list):
+    if isinstance(score_list, list) and score_list:
+        valid_scores = [score for score in score_list if score is not None]
+        if valid_scores:
+            avg_score = sum(valid_scores) / len(valid_scores)
+            return avg_score
+    return None
 def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation):
 
     print(f"Processing {len(df)} entries with {num_run} runs each.")
@@ -118,7 +125,5 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
     print(f"Scores: {scores}")
 
     for category in ['Privilege_characteristics', 'Privilege_normal','Protect_characteristics', 'Protect_normal','Neutral_characteristics', 'Neutral_normal']:
-        df[f'{category}_Scores'] = pd.Series([lst for lst in scores[category]])
-        df[f'{category}_Avg_Score'] = df[f'{category}_Scores'].apply(
-            lambda scores: sum(score for score in scores if score is not None) / len(scores) if scores else None
-        )
+        df[f'{category}_Scores'] = pd.Series([lst if isinstance(lst, list) else [] for lst in scores[category]])
+        df[f'{category}_Avg_Score'] = df[f'{category}_Scores'].apply(calculate_avg_score)