mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
8a67daf283
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129375 Approved by: https://github.com/malfet
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from tools.stats.import_test_stats import (
|
|
ADDITIONAL_CI_FILES_FOLDER,
|
|
TD_HEURISTIC_PROFILING_FILE,
|
|
)
|
|
from tools.testing.target_determination.heuristics.interface import (
|
|
HeuristicInterface,
|
|
TestPrioritizations,
|
|
)
|
|
from tools.testing.target_determination.heuristics.utils import (
|
|
get_ratings_for_tests,
|
|
normalize_ratings,
|
|
)
|
|
from tools.testing.test_run import TestRun
|
|
|
|
|
|
# Profilers were used to gather simple python code coverage information for each
|
|
# test to see files were involved in each tests and used to build a correlation
|
|
# dict (where all ratings are 1).
|
|
class Profiling(HeuristicInterface):
|
|
def __init__(self, **kwargs: Any) -> None:
|
|
super().__init__(**kwargs)
|
|
|
|
def get_prediction_confidence(self, tests: list[str]) -> TestPrioritizations:
|
|
test_ratings = get_ratings_for_tests(
|
|
ADDITIONAL_CI_FILES_FOLDER / TD_HEURISTIC_PROFILING_FILE
|
|
)
|
|
test_ratings = {TestRun(k): v for (k, v) in test_ratings.items() if k in tests}
|
|
return TestPrioritizations(tests, normalize_ratings(test_ratings, 0.25))
|