Commit
·
85ee43a
1
Parent(s):
75ece7c
Adds a progress bar hopefully.
Browse files
evaluate_genericify_cpp.py
CHANGED
|
@@ -10,6 +10,7 @@ import datasets
|
|
| 10 |
import evaluate
|
| 11 |
import itertools
|
| 12 |
import numpy as np
|
|
|
|
| 13 |
|
| 14 |
from .execute import check_correctness
|
| 15 |
|
|
@@ -158,7 +159,9 @@ class EvaluateGenericifyCpp(evaluate.Metric):
|
|
| 158 |
futures.append(future)
|
| 159 |
completion_id[task_id] += 1
|
| 160 |
|
| 161 |
-
for future in
|
|
|
|
|
|
|
| 162 |
result = future.result()
|
| 163 |
results[result["task_id"]].append(
|
| 164 |
(result["completion_id"], result)
|
|
|
|
| 10 |
import evaluate
|
| 11 |
import itertools
|
| 12 |
import numpy as np
|
| 13 |
+
from tqdm import tqdm
|
| 14 |
|
| 15 |
from .execute import check_correctness
|
| 16 |
|
|
|
|
| 159 |
futures.append(future)
|
| 160 |
completion_id[task_id] += 1
|
| 161 |
|
| 162 |
+
for future in tqdm(
|
| 163 |
+
concurrent.futures.as_completed(futures), desc="Evaluating"
|
| 164 |
+
):
|
| 165 |
result = future.result()
|
| 166 |
results[result["task_id"]].append(
|
| 167 |
(result["completion_id"], result)
|