frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[TD] More files for test_public_bindings (#132284)

Browse Source 
It relies on that file

Also we care about .cpp files too apparently
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132284
Approved by: https://github.com/ZainRizvi
This commit is contained in:
Catherine Lee
2024-07-31 19:53:39 +00:00
committed by PyTorch MergeBot
parent cb4c107d70
commit aeb78c9849
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/testing/target_determination/heuristics/public_bindings.py
View File

@ -15,6 +15,7 @@ class PublicBindings(HeuristicInterface):
# Literally just a heuristic for test_public_bindings. Pretty much anything
# that changes the public API can affect this testp
test_public_bindings = "test_public_bindings"
additional_files = ["test/allowlist_for_publicAPI.json"]
def __init__(self, **kwargs: dict[str, Any]) -> None:
super().__init__(**kwargs)
@ -28,7 +29,8 @@ class PublicBindings(HeuristicInterface):
changed_files = []
if any(
file.startswith("torch/") and file.endswith(".py") for file in changed_files
file.startswith("torch/") or file in self.additional_files
for file in changed_files
):
test_ratings[TestRun(self.test_public_bindings)] = 1.0
return TestPrioritizations(tests, test_ratings)