diff --git a/owl/run_gaia_roleplaying.py b/owl/run_gaia_roleplaying.py index cc08cd2..d375c4c 100644 --- a/owl/run_gaia_roleplaying.py +++ b/owl/run_gaia_roleplaying.py @@ -131,8 +131,8 @@ def main(): ) # Output results - logger.success(f"Correct: {result['correct']}, Total: {result['total']}") - logger.success(f"Accuracy: {result['accuracy']}") + logger.info(f"Correct: {result['correct']}, Total: {result['total']}") + logger.info(f"Accuracy: {result['accuracy']}") if __name__ == "__main__": diff --git a/owl/utils/gaia.py b/owl/utils/gaia.py index ec12ce6..a32bc1d 100644 --- a/owl/utils/gaia.py +++ b/owl/utils/gaia.py @@ -95,7 +95,7 @@ class GAIABenchmark(BaseBenchmark): json.dump(constructed_data, f, indent=4) f.close() - print(f"Successfully dumped tasks to {save_path}") + print(f"infofully dumped tasks to {save_path}") def load(self, force_download=False): r"""Load the GAIA dataset. @@ -195,7 +195,7 @@ class GAIABenchmark(BaseBenchmark): # Process tasks for task in tqdm(datas, desc="Running"): if self._check_task_completed(task["task_id"]): - logger.success( + logger.info( f"The following task is already completed:\n task id: {task['task_id']}, question: {task['Question']}" ) continue