Zekun Wu commited on
Commit
dee7dcf
1 Parent(s): 29abe81
Files changed (1) hide show
  1. util/injection.py +4 -3
util/injection.py CHANGED
@@ -36,7 +36,7 @@ def create_summary(row, group_name, label, occupation, additional_charateristics
36
 
37
 
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():
@@ -98,8 +98,9 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
98
  summary.append(f"{readable_name}: {value};")
99
 
100
  if additional_charateristics == True:
101
- additional_charateristics = [
102
- create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)]
 
103
 
104
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
105
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
 
36
 
37
 
38
 
39
+ summary.append(f"\n{additional_charateristics}")
40
 
41
  # for column, value in row.items():
42
  # if group_name.lower() not in column.lower():
 
98
  summary.append(f"{readable_name}: {value};")
99
 
100
  if additional_charateristics == True:
101
+ additional_charateristics = create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)
102
+ else:
103
+ additional_charateristics = ""
104
 
105
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
106
  prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)