xref: /aosp_15_r20/external/pytorch/tools/stats/export_test_times.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1import sys
2from pathlib import Path
3
4
5REPO_ROOT = Path(__file__).resolve().parent.parent.parent
6sys.path.append(str(REPO_ROOT))
7from tools.stats.import_test_stats import get_test_class_times, get_test_times
8
9
10def main() -> None:
11    print("Exporting test times from test-infra")
12    get_test_times()
13    get_test_class_times()
14
15
16if __name__ == "__main__":
17    main()
18