Wendong-Fan commited on
Commit
194e16b
·
2 Parent(s): 3993dd5 681f15d

fix logger bug (#241)

Browse files
Files changed (2) hide show
  1. owl/run_gaia_roleplaying.py +2 -2
  2. owl/utils/gaia.py +1 -1
owl/run_gaia_roleplaying.py CHANGED
@@ -131,8 +131,8 @@ def main():
131
  )
132
 
133
  # Output results
134
- logger.success(f"Correct: {result['correct']}, Total: {result['total']}")
135
- logger.success(f"Accuracy: {result['accuracy']}")
136
 
137
 
138
  if __name__ == "__main__":
 
131
  )
132
 
133
  # Output results
134
+ logger.info(f"Correct: {result['correct']}, Total: {result['total']}")
135
+ logger.info(f"Accuracy: {result['accuracy']}")
136
 
137
 
138
  if __name__ == "__main__":
owl/utils/gaia.py CHANGED
@@ -195,7 +195,7 @@ class GAIABenchmark(BaseBenchmark):
195
  # Process tasks
196
  for task in tqdm(datas, desc="Running"):
197
  if self._check_task_completed(task["task_id"]):
198
- logger.success(
199
  f"The following task is already completed:\n task id: {task['task_id']}, question: {task['Question']}"
200
  )
201
  continue
 
195
  # Process tasks
196
  for task in tqdm(datas, desc="Running"):
197
  if self._check_task_completed(task["task_id"]):
198
+ logger.info(
199
  f"The following task is already completed:\n task id: {task['task_id']}, question: {task['Question']}"
200
  )
201
  continue