def cluster(jobs, master=None, name=None, quiet=False): if isinstance(jobs, dict): jobs = [Job(**jobs)] if isinstance(jobs, Job): jobs = [jobs] jobs = [job if isinstance(job, Job) else Job(**job) for job in jobs] s = TFMesosScheduler(jobs, master=master, name=name, quiet=quiet) yield s.start() s.stop()
def cluster(jobs, **kw): if isinstance(jobs, dict): jobs = [Job(**jobs)] if isinstance(jobs, Job): jobs = [jobs] jobs = [job if isinstance(job, Job) else Job(**job) for job in jobs] s = TFMesosScheduler(jobs, **kw) try: s.start() yield s finally: s.stop()
def cluster(jobs, master=None, name=None, quiet=False, volumes={}, local_task=None, containerizer_type=None): if isinstance(jobs, dict): jobs = [Job(**jobs)] if isinstance(jobs, Job): jobs = [jobs] jobs = [job if isinstance(job, Job) else Job(**job) for job in jobs] s = TFMesosScheduler(jobs, master=master, name=name, quiet=quiet, volumes=volumes, local_task=local_task, containerizer_type=containerizer_type) yield s.start() s.stop()