From 34ad77b841601bf73d612208b52773c76cec03b3 Mon Sep 17 00:00:00 2001
From: Jeremy Reizenstein <reizenstein@fb.com>
Date: Wed, 7 Sep 2022 14:21:15 -0700
Subject: [PATCH] trainer test completion

Summary: allow TESTIT to complete properly

Reviewed By: kjchalup

Differential Revision: D39280546

fbshipit-source-id: 38fe69988a736e32dbe78d1d05e6d8421353854a
---
 dev/test_list.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dev/test_list.py b/dev/test_list.py
index 65d95f22f..f00facd8d 100644
--- a/dev/test_list.py
+++ b/dev/test_list.py
@@ -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]: