jjkim commited on
Commit
655bbad
1 Parent(s): 9328e33

put disable_tqdm option

Browse files
Files changed (1) hide show
  1. 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: