Zekun Wu commited on
Commit
8808aa8
1 Parent(s): 2abe0cf
Files changed (1) hide show
  1. util/injection.py +2 -2
util/injection.py CHANGED
@@ -36,7 +36,7 @@ def create_summary(row, group_name, label, occupation, additional_charateristics
36
 
37
  summary.append(info)
38
 
39
- summary.append("\n" + additional_charateristics)
40
 
41
  # for column, value in row.items():
42
  # if group_name.lower() not in column.lower():
@@ -96,7 +96,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
96
 
97
  if additional_charateristics == True:
98
  additional_charateristics = [
99
- create_charateristics("\n".join(summary), group_name, occupation, agent, parameters)]
100
 
101
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
102
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
 
36
 
37
  summary.append(info)
38
 
39
+ summary.append("\n" + additional_charateristics) if additional_charateristics else ""
40
 
41
  # for column, value in row.items():
42
  # if group_name.lower() not in column.lower():
 
96
 
97
  if additional_charateristics == True:
98
  additional_charateristics = [
99
+ create_charateristics(row['resume'], group_name, occupation, agent, parameters)]
100
 
101
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
102
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)