Update openAI_manager.py

openAI_manager.py

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