def stage(op):
run = init_run(op)
try:
_stage_run_proc_env(op, run)
_resolve_deps(op, run, for_stage=True)
op_util.set_run_staged(run)
finally:
op_util.clear_run_pending(run)
return run
def _init_trial_run(batch_run, trial_flag_vals, run_dir=None):
run = op_util.init_run(run_dir)
_link_to_trial(batch_run, run)
proto_run = batch_run.batch_proto
util.copytree(proto_run.dir, run.dir)
run.write_attr("flags", trial_flag_vals)
run.write_attr("label", _trial_label(proto_run, trial_flag_vals))
run.write_attr("op", _trial_op_attr(proto_run, trial_flag_vals))
op_util.set_run_staged(run)
return run
def stage(op, continue_on_deps_error=False):
run = init_run(op)
try:
_stage_run_proc_env(op, run)
_resolve_deps(op,
run,
for_stage=True,
continue_on_error=continue_on_deps_error)
op_util.set_run_staged(run)
finally:
op_util.clear_run_pending(run)
return run
def init_trial_run(batch_run, trial_flag_vals, run_dir=None):
run = op_util.init_run(run_dir)
_link_to_trial(batch_run, run)
proto_run = batch_run.batch_proto
assert proto_run, "proto_run not initialized for batch %s (%s)" % (
batch_run.id,
batch_run.dir,
)
util.copytree(proto_run.dir, run.dir)
run.write_attr("flags", trial_flag_vals)
run.write_attr("label", _trial_label(proto_run, trial_flag_vals))
run.write_attr("op", _trial_op_attr(proto_run, trial_flag_vals))
op_util.set_run_staged(run)
return run
