Zekun Wu commited on
Commit
fd2d5d2
1 Parent(s): f91351f
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -100,7 +100,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
100
  charateristics = ""
101
 
102
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
103
- prompt_temp = create_summary(row, group_name, label, occupation, charateristics)
104
  print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
105
  print("=============================================================")
106
  result = invoke_retry(prompt_temp, agent, parameters)
 
100
  charateristics = ""
101
 
102
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
103
+ prompt_temp = create_summary(group_name, label, occupation, charateristics)
104
  print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
105
  print("=============================================================")
106
  result = invoke_retry(prompt_temp, agent, parameters)