Skip to content

Commit

Permalink
trainer test completion
Browse files Browse the repository at this point in the history
Summary: allow TESTIT to complete properly

Reviewed By: kjchalup

Differential Revision: D39280546

fbshipit-source-id: 38fe69988a736e32dbe78d1d05e6d8421353854a
  • Loading branch information
bottler authored and facebook-github-bot committed Sep 7, 2022
1 parent 90b758f commit 34ad77b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dev/test_list.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,8 @@

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]:
Expand Down

0 comments on commit 34ad77b

Please sign in to comment.