rjiang12 commited on
Commit
dc2921c
·
1 Parent(s): 206ff8d

Update openAI_manager.py

Browse files
Files changed (1) hide show
  1. openAI_manager.py +2 -2
openAI_manager.py CHANGED
@@ -88,8 +88,8 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
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
- print(f"{((gi*len(att)+ai)*len(gens)+gen_i)/(num_steps*len(gens))}: {gen[2]}")
92
- progress(((gi*len(att)+ai)*len(gens)+gen_i)/(num_steps*len(gens)), desc=f"{gen[2]}")
93
  all_gens.extend(gens)
94
 
95
  return all_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
+ print(f"{((gi*len(attribute_kwds)+ai)*len(gens)+gen_i)/(num_steps*len(gens))}: {gen[2]}")
92
+ progress(((gi*len(attribute_kwds)+ai)*len(gens)+gen_i)/(num_steps*len(gens)), desc=f"{gen[2]}")
93
  all_gens.extend(gens)
94
 
95
  return all_gens