def default_main(seq_trial_cb, non_repeating=True): if non_repeating: seq_trial_cb = NonRepeatingTrials(seq_trial_cb) try: batch_util.seq_trials_main(seq_trial_cb, State) except batch_util.StopBatch as e: assert e.error sys.exit(1)
def default_main(seq_trial_cb, non_repeating=True): _patch_numpy_deprecation_warnings() if non_repeating: seq_trial_cb = NonRepeatingTrials(seq_trial_cb) batch_util.seq_trials_main(State, seq_trial_cb)
import sys from guild import batch_util batch_fail = False trials_fail = "" def init_state(batch): return batch_util.SeqState(batch, trial=0, trials_fail=_trials_fail()) def _trials_fail(): return [int(s) for s in str(trials_fail).split(",") if s] def init_trial(trial, state): flags = {} flags.update(state.proto_flags) if state.trial in state.trials_fail: if batch_fail: print("BATCH FAIL") sys.exit(1) flags["fail"] = True state.trial += 1 return flags, {} batch_util.seq_trials_main(init_trial, init_state)