rjiang12 commited on
Commit
90b8d9f
·
1 Parent(s): 36d6f7f

Update openAI_manager.py

Browse files
Files changed (1) hide show
  1. openAI_manager.py +2 -1
openAI_manager.py CHANGED
@@ -78,12 +78,13 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
78
  num_steps = len(group_kwds)*len(attribute_kwds)
79
  for gi, grp_kwd in enumerate(group_kwds):
80
  for ai, att_kwd in enumerate(attribute_kwds):
81
- progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"Generating {grp_kwd}<>{att_kwd}...")
82
 
83
  kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
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
  all_gens.extend(gens)
88
 
89
  return all_gens
 
78
  num_steps = len(group_kwds)*len(attribute_kwds)
79
  for gi, grp_kwd in enumerate(group_kwds):
80
  for ai, att_kwd in enumerate(attribute_kwds):
81
+ # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"Generating {grp_kwd}<>{att_kwd}...")
82
 
83
  kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
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