def main():
jobs_dir = os.path.join(SCRIPT_DIR, 'jobs')
seed_rng = random.SystemRandom()
if os.path.exists(jobs_dir):
sys.stderr.write('{} already exists; aborting.\n'.format(jobs_dir))
sys.exit(1)
if not os.path.exists(SIM_DB_PATH):
sys.stderr.write('simulations DB not present; aborting.\n')
sys.exit(1)
# Make sure uzal costfunc binary has been built
sys.stderr.write('Ensuring costfunc binary has been built...\n')
uzalcost.set_up_uzal_costfunc()
with sqlite3.connect(SIM_DB_PATH) as db:
for job_id, job_subdir, eps, beta00, sigma01, sd_proc, replicate_id, random_seed in db.execute(
'SELECT * FROM job_info WHERE replicate_id < ?',
[N_REPLICATES]):
job_dir = os.path.join(SCRIPT_DIR, job_subdir)
sys.stderr.write('{0}\n'.format(job_dir))
os.makedirs(job_dir)
runmany_info = OrderedDict(RUNMANY_INFO_TEMPLATE)
runmany_info['simulation_job_id'] = job_id
runmany_info['job_info'] = OrderedDict([
('eps', eps), ('beta00', beta00), ('sigma01', sigma01),
('sd_proc', sd_proc), ('replicate_id', replicate_id),
('random_seed', seed_rng.randint(1, 2**31 - 1))
])
dump_json(runmany_info, os.path.join(job_dir, 'runmany_info.json'))
def main():
jobs_dir = os.path.join(SCRIPT_DIR, 'jobs')
seed_rng = random.SystemRandom()
if os.path.exists(jobs_dir):
sys.stderr.write('{} already exists; aborting.\n'.format(jobs_dir))
sys.exit(1)
if not os.path.exists(SIM_DB_PATH):
sys.stderr.write('simulations DB not present; aborting.\n')
sys.exit(1)
# Make sure uzal costfunc binary has been built
sys.stderr.write('Ensuring costfunc binary has been built...\n')
uzalcost.set_up_uzal_costfunc()
with sqlite3.connect(SIM_DB_PATH) as db:
for job_id, job_subdir, eps, beta00, sigma01, sd_proc, replicate_id, random_seed in db.execute(
'SELECT * FROM job_info WHERE replicate_id < ?', [N_REPLICATES]
):
job_dir = os.path.join(SCRIPT_DIR, job_subdir)
sys.stderr.write('{0}\n'.format(job_dir))
os.makedirs(job_dir)
runmany_info = OrderedDict(RUNMANY_INFO_TEMPLATE)
runmany_info['simulation_job_id'] = job_id
runmany_info['job_info'] = OrderedDict([
('eps', eps),
('beta00', beta00),
('sigma01', sigma01),
('sd_proc', sd_proc),
('replicate_id', replicate_id),
('random_seed', seed_rng.randint(1, 2**31-1))
])
dump_json(runmany_info, os.path.join(job_dir, 'runmany_info.json'))