trainer test completion

Summary: allow TESTIT to complete properly

Reviewed By: kjchalup

Differential Revision: D39280546

fbshipit-source-id: 38fe69988a736e32dbe78d1d05e6d8421353854a

dev/test_list.py

@@ -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]: