def runscript(): parser = argparse.ArgumentParser() for arg in arguments: parser.add_argument(*arg['flags'], **arg['parameters']) run_args = parser.parse_args() pbs_vars = cli.set_env_vars(run_args.init_run, run_args.n_runs, run_args.lab_path, run_args.dir_path) for var in pbs_vars: os.environ[var] = str(pbs_vars[var]) lab = Laboratory(run_args.model_type, run_args.config_path, run_args.lab_path) expt = Experiment(lab) if 'PBS_NCPUS' not in os.environ: # Not a PBS batch job: set ncpus in environment if 'collate_ncpus' in expt.config: os.environ['NCPUS'] = str(expt.config['collate_ncpus']) expt.collate() if expt.postscript: expt.postprocess()
def runscript(): parser = argparse.ArgumentParser() for arg in arguments: parser.add_argument(*arg['flags'], **arg['parameters']) run_args = parser.parse_args() pbs_vars = cli.set_env_vars(run_args.init_run, run_args.n_runs, run_args.lab_path, run_args.dir_path) for var in pbs_vars: os.environ[var] = str(pbs_vars[var]) lab = Laboratory(run_args.model_type, run_args.config_path, run_args.lab_path) expt = Experiment(lab) expt.collate() if expt.postscript: expt.postprocess()