jjkim
commited on
Commit
·
655bbad
1
Parent(s):
9328e33
put disable_tqdm option
Browse files- code_eval.py +2 -1
code_eval.py
CHANGED
|
@@ -163,6 +163,7 @@ class CodeEval(evaluate.Metric):
|
|
| 163 |
num_workers=4,
|
| 164 |
timeout=3.0,
|
| 165 |
early_stop=False,
|
|
|
|
| 166 |
):
|
| 167 |
"""Returns the scores"""
|
| 168 |
|
|
@@ -190,7 +191,7 @@ class CodeEval(evaluate.Metric):
|
|
| 190 |
result.add(future)
|
| 191 |
results[tid].append(result)
|
| 192 |
|
| 193 |
-
pbar = tqdm(total=sum(len(r) for r in results.values()))
|
| 194 |
prev_done_count = 0
|
| 195 |
done = False
|
| 196 |
while not done:
|
|
|
|
| 163 |
num_workers=4,
|
| 164 |
timeout=3.0,
|
| 165 |
early_stop=False,
|
| 166 |
+
disable_tqdm=False,
|
| 167 |
):
|
| 168 |
"""Returns the scores"""
|
| 169 |
|
|
|
|
| 191 |
result.add(future)
|
| 192 |
results[tid].append(result)
|
| 193 |
|
| 194 |
+
pbar = tqdm(total=sum(len(r) for r in results.values()), disable=disable_tqdm)
|
| 195 |
prev_done_count = 0
|
| 196 |
done = False
|
| 197 |
while not done:
|