Zekun Wu commited on
Commit
d360cd7
1 Parent(s): 9092787
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -123,7 +123,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
123
  df[f'{category}_{key}_Scores'] = pd.Series([lst for lst in scores[f'{category}_{key}']])
124
  df[f'{category}_{key}_Avg_Score'] = df[f'{category}_{key}_Scores'].apply(
125
  lambda score_list: sum(score for score in score_list if score not in [None, -1]) / len(score_list)
126
- if score_list else None
127
  )
128
 
129
  # Debugging: Check the calculated average scores
 
123
  df[f'{category}_{key}_Scores'] = pd.Series([lst for lst in scores[f'{category}_{key}']])
124
  df[f'{category}_{key}_Avg_Score'] = df[f'{category}_{key}_Scores'].apply(
125
  lambda score_list: sum(score for score in score_list if score not in [None, -1]) / len(score_list)
126
+ if isinstance(score_list, list) and score_list else None
127
  )
128
 
129
  # Debugging: Check the calculated average scores