callanwu commited on
Commit
20bb6a0
·
1 Parent(s): 037cb58

fix logger bug

Browse files
Files changed (2) hide show
  1. owl/run_gaia_roleplaying.py +2 -2
  2. owl/utils/gaia.py +2 -2
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
@@ -95,7 +95,7 @@ class GAIABenchmark(BaseBenchmark):
95
  json.dump(constructed_data, f, indent=4)
96
  f.close()
97
 
98
- print(f"Successfully dumped tasks to {save_path}")
99
 
100
  def load(self, force_download=False):
101
  r"""Load the GAIA dataset.
@@ -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
 
95
  json.dump(constructed_data, f, indent=4)
96
  f.close()
97
 
98
+ print(f"infofully dumped tasks to {save_path}")
99
 
100
  def load(self, force_download=False):
101
  r"""Load the GAIA dataset.
 
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