xref: /aosp_15_r20/external/pytorch/tools/testing/target_determination/gen_artifact.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1from __future__ import annotations
2
3import json
4import os
5from pathlib import Path
6from typing import Any
7
8
9REPO_ROOT = Path(__file__).resolve().parent.parent.parent.parent
10
11
12def gen_ci_artifact(included: list[Any], excluded: list[Any]) -> None:
13    file_name = f"td_exclusions-{os.urandom(10).hex()}.json"
14    with open(REPO_ROOT / "test" / "test-reports" / file_name, "w") as f:
15        json.dump({"included": included, "excluded": excluded}, f)
16