update

util/injection.py

@@ -124,8 +124,31 @@ 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']:
+    for category in ['Privilege_characteristics', 'Privilege_normal', 'Protect_characteristics', 'Protect_normal',
+                     'Neutral_characteristics', 'Neutral_normal']:
+        # Debug: Print the scores for the current category
+        print(f"Processing category: {category}")
+        print(f"Scores: {scores[category]}")
+
+        # Ensure the scores are lists
         df[f'{category}_Scores'] = pd.Series([lst if isinstance(lst, list) else [] for lst in scores[category]])
+
+        # Debug: Check the Series after assignment
+        print(f"Series for {category}_Scores:\n{df[f'{category}_Scores']}")
+
+        # Calculate the average score with additional debug info
+        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)
+                    print(f"Valid scores: {valid_scores}, Average score: {avg_score}")
+                    return avg_score
+            return None
+
         df[f'{category}_Avg_Score'] = df[f'{category}_Scores'].apply(calculate_avg_score)
 
+        # Debug: Print the calculated average scores
+        print(f"Average scores for {category}:\n{df[f'{category}_Avg_Score']}")
+
     return df