1

Signed-off-by: kit <101046518@qq.com>
geekan · Oct 27, 2024 · a7fa56a · a7fa56a
1 parent ea699ca
commit a7fa56a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py
@@ -28,7 +28,7 @@ async def main():
             predictions.append(str(""))
     df = pd.DataFrame({"Label": labels, "Prediction": predictions, "T/F": is_true})
     df.to_excel("DABench_output.xlsx", index=False)
-    print(DA.eval_all(id_list, predictions))
+    logger.info(DA.eval_all(id_list, predictions))
 
 
 if __name__ == "__main__":

diff --git a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py
@@ -15,7 +15,7 @@ async def main(id=0):
     logger.info(result)
     save_history(role=di)
     _, is_correct = DA.eval(id, str(result))
-    print(f"Prediction is {'correct' if is_correct else 'incorrect'}.")
+    logger.info(f"Prediction is {'correct' if is_correct else 'incorrect'}.")
 
 
 if __name__ == "__main__":