jjkim
commited on
Commit
•
029f252
1
Parent(s):
fa82c7f
fix pbar total count
Browse files- code_eval.py +1 -1
code_eval.py
CHANGED
|
@@ -190,7 +190,7 @@ class CodeEval(evaluate.Metric):
|
|
| 190 |
result.add(future)
|
| 191 |
results[tid].append(result)
|
| 192 |
|
| 193 |
-
pbar = tqdm(total=len(results))
|
| 194 |
prev_done_count = 0
|
| 195 |
done = False
|
| 196 |
while not done:
|
|
|
|
| 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:
|