Zekun Wu commited on
Commit
9edca9e
1 Parent(s): 7b1eae1
Files changed (1) hide show
  1. util/injection.py +2 -0
util/injection.py CHANGED
@@ -87,6 +87,8 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
87
 
88
  def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation,
89
  additional_charateristics):
 
 
90
  """ Process entries and compute scores concurrently, with progress updates. """
91
  scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
92
 
 
87
 
88
  def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation,
89
  additional_charateristics):
90
+
91
+ print(f"Processing {len(df)} entries with {num_run} runs each.")
92
  """ Process entries and compute scores concurrently, with progress updates. """
93
  scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
94