Update openAI_manager.py

openAI_manager.py

@@ -85,9 +85,8 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
         gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
         # print(f"Gens for pair: <{kwd_pair}> -> {gens}")
         # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
-        for gen in gens:
-            print(gen)
-            progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gen[2]}")
+        for gen_i, gen in enumerate(gens):
+            progress((gi*len(attribute_kwds)+((gen_i + 1)/len(gens)*ai))/num_steps, desc=f"{gen[2]}")
         all_gens.extend(gens)
 
     return all_gens