Zekun Wu commited on
Commit
a2c85d8
1 Parent(s): 774346d
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -123,7 +123,7 @@ def process_scores_single(df, num_run, parameters, counterfactual_label, agent,
123
  for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
124
  for key, label in zip(['Counterfactual', 'Neutral'], [counterfactual_label, False]):
125
  prompt_temp = create_summary(row, group_name, label, occupation)
126
- print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
127
  print("=============================================================")
128
  result = invoke_retry(prompt_temp, agent, parameters)
129
  scores[key][index].append(result)
 
123
  for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
124
  for key, label in zip(['Counterfactual', 'Neutral'], [counterfactual_label, False]):
125
  prompt_temp = create_summary(row, group_name, label, occupation)
126
+ print(f"Run {run + 1} - Entry {index + 1} - {key}")
127
  print("=============================================================")
128
  result = invoke_retry(prompt_temp, agent, parameters)
129
  scores[key][index].append(result)