rjiang12 commited on
Commit
2394976
·
1 Parent(s): f90e9ba

Update openAI_manager.py

Browse files
Files changed (1) hide show
  1. openAI_manager.py +1 -0
openAI_manager.py CHANGED
@@ -86,6 +86,7 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
86
  # print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
  # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
88
  for gen_i, gen in enumerate(gens):
 
89
  progress((gi*len(attribute_kwds)+((gen_i + 1)/len(gens)*ai))/num_steps, desc=f"{gen[2]}")
90
  all_gens.extend(gens)
91
 
 
86
  # print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
  # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
88
  for gen_i, gen in enumerate(gens):
89
+ print(f"{gi*len(attribute_kwds)+((gen_i + 1)/len(gens)*ai}: {gen[2]}")
90
  progress((gi*len(attribute_kwds)+((gen_i + 1)/len(gens)*ai))/num_steps, desc=f"{gen[2]}")
91
  all_gens.extend(gens)
92