in summarize_from_feedback/utils/jobs.py [0:0]
def launch(job: Job):
H = job.params
if H.mode == "local":
with open("/tmp/pickle_fn", "wb") as file:
cloudpickle.dump(job.fn, file)
env = os.environ.copy()
env["JOB_NAME"] = H.name
subprocess.check_call(
[
"mpiexec",
"-n",
str(H.mpi),
"python",
"-c",
'import sys; import pickle; pickle.loads(open("/tmp/pickle_fn", "rb").read())()',
],
env=env,
)
else:
raise NotImplementedError(f"unsupported launch mode {H.mode}")