in launcher/nemo/stages.py [0:0]
def _make_docker_exec_script_text(self, stage_cfg_path):
docker_exec_script_text = ["#!/bin/bash", "set -ex"]
docker_exec_script_text.append("")
docker_exec_script_text.append("function job_epilogue {")
docker_exec_script_text.append(
" docker ps -a --filter 'name="
+ CONTAINER_NAME
+ "' --format '{{.ID}}' | xargs -I{} docker rm -f {} > /dev/null 2>&1 || true"
)
docker_exec_script_text.append("}")
docker_exec_script_text.append("trap job_epilogue EXIT SIGTERM SIGINT")
docker_exec_script_text.append("")
docker_exec_script_text.append(f"docker exec {CONTAINER_NAME} bash {stage_cfg_path.parents[0]}/train_script.sh")
docker_exec_script_text.append("")
docker_exec_script_text.append("exit 0")
return "\n".join(docker_exec_script_text)