red1bluelost commited on
Commit
85ee43a
1 Parent(s): 75ece7c

Adds a progress bar hopefully.

Browse files
Files changed (1) hide show
  1. evaluate_genericify_cpp.py +4 -1
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 concurrent.futures.as_completed(futures):
 
 
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)