fix logger bug (#241)
Browse files- owl/run_gaia_roleplaying.py +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.
|
135 |
-
logger.
|
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.
|
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
|