update plotting, add overwrite option to sens.py

1 files changed, 9 insertions, 0 deletions

diff --git a/sens.py b/sens.py
index 74eef61..9e58c2f 100755
--- a/sens.py
+++ b/sens.py
@@ -167,6 +167,10 @@ def parse_args(args=None):
         '--eval-segment', type=str, default='all',
         help='Which point to evalaute'
     )
+    parser.add_argument(
+        '--overwrite', type=misc_utils.parse_bool, default='False',
+        help='Overwrite chains'
+    )
     fr_utils.fr_argparse(parser)
     gf_utils.gf_argparse(parser)
     llh_utils.llh_argparse(parser)
@@ -204,6 +208,11 @@ def main():
         *map(misc_utils.parse_enum, [args.stat_method, args.data])
     ) + misc_utils.gen_identifier(args)
 
+    if not args.overwrite and os.path.isfile(outfile+'.npy'):
+        print 'FILE EXISTS {0}'.format(outfile+'.npy')
+        print 'Exiting...'
+        return
+
     # Setup Golemfit.
     if args.run_mn:
         gf_utils.setup_fitter(args, asimov_paramset)