Skip to content

Commit ceefe7a

Browse files
committed
Fix a few more bugs
1 parent 6115ca4 commit ceefe7a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

runPerformanceTests.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ def read_tests(filename, default_num_samples):
3636
num_samples_list = []
3737
with open(filename) as f:
3838
for line in f:
39+
line = line.strip()
3940
if line.startswith("#"): continue
4041
if ", " in line:
4142
model, num_samples = line.split(", ")
@@ -339,8 +340,8 @@ def process_test_wrapper(tup):
339340

340341
executables = [m[:-5] for m in models]
341342
make_time, _ = time_step("make_all_models", make, executables, args.j)
342-
tests = [(model, exe, find_data_for_model(model), num_samples)
343-
for model, exe, num_samples in zip(models, executables, num_samples)]
343+
tests = [(model, exe, find_data_for_model(model), ns)
344+
for model, exe, ns in zip(models, executables, num_samples)]
344345
if args.runj > 1:
345346
tp = ThreadPool(args.runj)
346347
map_ = tp.imap_unordered

0 commit comments

Comments
 (0)