pytorch

1
import pathlib
2
import sys
3

4
REPO_ROOT = pathlib.Path(__file__).resolve().parent.parent.parent
5
sys.path.append(str(REPO_ROOT))
6
from tools.stats.import_test_stats import get_test_class_times, get_test_times
7

8

9
def main() -> None:
10
    print("Exporting test times from test-infra")
11
    get_test_times()
12
    get_test_class_times()
13

14

15
if __name__ == "__main__":
16
    main()
17