rjiang12 commited on
Commit
471a25e
·
1 Parent(s): 90b8d9f

Update openAI_manager.py

Browse files
Files changed (1) hide show
  1. openAI_manager.py +3 -1
openAI_manager.py CHANGED
@@ -84,7 +84,9 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
84
 
85
  gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
86
  print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
- progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
 
 
88
  all_gens.extend(gens)
89
 
90
  return all_gens
 
84
 
85
  gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
86
  print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
+ # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
88
+ for gen in gens:
89
+ progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gen}")
90
  all_gens.extend(gens)
91
 
92
  return all_gens