in heyhi/__init__.py [0:0]
def kill_and_prune(self, silent: bool) -> None:
if not self.exp_path.exists():
return
logging.info("Prune+kill for %s", self.exp_path)
if not silent:
print("Deleting the folder in 3 seconds", end="", flush=True)
for _ in range(3):
print(".", end="", flush=True)
time.sleep(1)
maybe_jobid = self.maybe_get_job_id()
if maybe_jobid is not None and maybe_jobid != LOCAL_JOB_ID:
if not silent:
print("killing job", maybe_jobid, "...", end="", flush=True)
subprocess.check_call(["scancel", str(maybe_jobid)])
if not silent:
print(" purging the log dir", "...", end="", flush=True)
shutil.rmtree(str(self.exp_path))
if not silent:
print("done")