mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-10-29 14:10:16 +08:00
yyj
a3695f1a83
* ci(release): prepare source builds from four main snapshots * docs(release): add verified Kimi RAWINT4 LoRA smoke configs * ci(release): gate four-main PyPI promotion on immutable wheels and model E2E
29 lines
1002 B
Python
29 lines
1002 B
Python
"""Standard-library-only cleanup, including when build dependency installation fails."""
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
|
|
def cleanup(directory, parent):
|
|
directory, parent = Path(directory), Path(parent).resolve()
|
|
if directory.is_symlink():
|
|
raise ValueError("Refuse symlink cleanup")
|
|
resolved = directory.resolve(strict=True)
|
|
if resolved.parent != parent or not resolved.name.startswith("kt-four-main."):
|
|
raise ValueError("Refuse broad cleanup")
|
|
if (resolved / ".kt-release-owned").read_text().strip() != os.environ[
|
|
"GITHUB_RUN_ID"
|
|
]:
|
|
raise ValueError("Build directory is not owned by this run")
|
|
shutil.rmtree(resolved)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--directory", type=Path, required=True)
|
|
parser.add_argument("--parent", type=Path, required=True)
|
|
args = parser.parse_args()
|
|
cleanup(args.directory, args.parent)
|