diff options
Diffstat (limited to 'run_exps.py')
-rwxr-xr-x | run_exps.py | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/run_exps.py b/run_exps.py index b9cf88e..6531415 100755 --- a/run_exps.py +++ b/run_exps.py | |||
@@ -8,12 +8,11 @@ import re | |||
8 | import shutil | 8 | import shutil |
9 | import sys | 9 | import sys |
10 | import run.tracer as trace | 10 | import run.tracer as trace |
11 | import traceback | ||
12 | 11 | ||
13 | from collections import namedtuple | 12 | from collections import namedtuple |
14 | from optparse import OptionParser | 13 | from optparse import OptionParser |
15 | from run.executable.executable import Executable | 14 | from run.executable.executable import Executable |
16 | from run.experiment import Experiment,ExperimentDone | 15 | from run.experiment import Experiment,ExperimentDone,ExperimentFailed,SystemCorrupted |
17 | from run.proc_entry import ProcEntry | 16 | from run.proc_entry import ProcEntry |
18 | 17 | ||
19 | '''Customizable experiment parameters''' | 18 | '''Customizable experiment parameters''' |
@@ -330,6 +329,7 @@ def main(): | |||
330 | created = True | 329 | created = True |
331 | os.mkdir(out_base) | 330 | os.mkdir(out_base) |
332 | 331 | ||
332 | ran = 0 | ||
333 | done = 0 | 333 | done = 0 |
334 | succ = 0 | 334 | succ = 0 |
335 | failed = 0 | 335 | failed = 0 |
@@ -362,15 +362,23 @@ def main(): | |||
362 | invalid += 1 | 362 | invalid += 1 |
363 | print("Invalid environment for experiment '%s'" % exp) | 363 | print("Invalid environment for experiment '%s'" % exp) |
364 | print(e) | 364 | print(e) |
365 | except: | 365 | except KeyboardInterrupt: |
366 | print("Keyboard interrupt, quitting") | ||
367 | break | ||
368 | except SystemCorrupted as e: | ||
369 | print("System is corrupted! Fix state before continuing.") | ||
370 | print(e) | ||
371 | break | ||
372 | except ExperimentFailed: | ||
366 | print("Failed experiment %s" % exp) | 373 | print("Failed experiment %s" % exp) |
367 | traceback.print_exc() | ||
368 | failed += 1 | 374 | failed += 1 |
369 | 375 | ||
376 | ran += 1 | ||
377 | |||
370 | if not os.listdir(out_base) and created and not succ: | 378 | if not os.listdir(out_base) and created and not succ: |
371 | os.rmdir(out_base) | 379 | os.rmdir(out_base) |
372 | 380 | ||
373 | message = "Experiments run:\t%d" % len(args) +\ | 381 | message = "Experiments ran:\t%d of %d" % (ran, len(args)) +\ |
374 | "\n Successful:\t\t%d" % succ +\ | 382 | "\n Successful:\t\t%d" % succ +\ |
375 | "\n Failed:\t\t%d" % failed +\ | 383 | "\n Failed:\t\t%d" % failed +\ |
376 | "\n Already Done:\t\t%d" % done +\ | 384 | "\n Already Done:\t\t%d" % done +\ |