trainer test completion

Summary: allow TESTIT to complete properly

Reviewed By: kjchalup

Differential Revision: D39280546

fbshipit-source-id: 38fe69988a736e32dbe78d1d05e6d8421353854a
This commit is contained in:
Jeremy Reizenstein 2022-09-07 14:21:15 -07:00 committed by Facebook GitHub Bot
parent 90b758f725
commit 34ad77b841

View File

@ -17,7 +17,8 @@ It has no dependencies.
def get_test_files() -> List[Path]:
root = Path(__file__).parent.parent
return list((root / "tests").glob("**/test*.py"))
dirs = ["tests", "projects/implicitron_trainer"]
return [i for dir in dirs for i in (root / dir).glob("**/test*.py")]
def tests_from_file(path: Path, base: str) -> List[str]: