async def async_run(self, job_name, **kwargs):
self._check_status(self.SessionStatus.RUNNING)
self._run_push_jobs(**kwargs)
job_inst = job_instance_util.MakeUserJobInstance(job_name)
self._run_job(job_inst)
output_futures = tuple(self._run_pull_jobs(job_name).values())
return await asyncio.gather(*output_futures)
def LaunchUserJob(self, job_func, *arg):
assert self.status_ is SessionStatus.RUNNING
job_name = job_func.__name__
push_util.AsyncPush(self, job_func, *arg)
self.LaunchJob(job_instance_util.MakeUserJobInstance(job_name))
return job_func.__oneflow_output_remote_blobs__
