Zekun Wu commited on
Commit
4e735ec
1 Parent(s): 0516ae0
Files changed (1) hide show
  1. util/injection.py +1 -3
util/injection.py CHANGED
@@ -20,8 +20,6 @@ def create_charateristics(original_resume, group_name, occupation, agent, parame
20
 
21
  additional_charateristics = invoke_retry(prompt, agent, parameters)
22
 
23
- print(f"Additional characteristics: {additional_charateristics}")
24
-
25
  combined_charateristics = f"{original_resume}\n{additional_charateristics}"
26
 
27
  return combined_charateristics
@@ -106,7 +104,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
106
 
107
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
108
  prompt_temp = create_summary(row, group_name, label, occupation, charateristics)
109
- print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{charateristics}")
110
  print("=============================================================")
111
  result = invoke_retry(prompt_temp, agent, parameters)
112
  scores[key][index].append(result)
 
20
 
21
  additional_charateristics = invoke_retry(prompt, agent, parameters)
22
 
 
 
23
  combined_charateristics = f"{original_resume}\n{additional_charateristics}"
24
 
25
  return combined_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, charateristics)
107
+ print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
108
  print("=============================================================")
109
  result = invoke_retry(prompt_temp, agent, parameters)
110
  scores[key][index].append(result)