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:
|