fix logger bug
Browse files- owl/run_gaia_roleplaying.py +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.
|
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
@@ -95,7 +95,7 @@ class GAIABenchmark(BaseBenchmark):
|
|
95 |
json.dump(constructed_data, f, indent=4)
|
96 |
f.close()
|
97 |
|
98 |
-
print(f"
|
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.
|
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
|