From f1924854f1486585fd34670c92b4c6665655c95c Mon Sep 17 00:00:00 2001
From: Stuart Siegel <12914116+ssiegel95@users.noreply.github.com>
Date: Tue, 7 Jan 2025 17:39:39 -0500
Subject: [PATCH] easier to just use an include list at this point

---
 services/inference/tests/test_inference_lib.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/services/inference/tests/test_inference_lib.py b/services/inference/tests/test_inference_lib.py
index 7b96898e..6119781c 100644
--- a/services/inference/tests/test_inference_lib.py
+++ b/services/inference/tests/test_inference_lib.py
@@ -29,13 +29,7 @@
 # SERIES_LENGTH = 512
 FORECAST_LENGTH = 96
 
-
-MODEL_IDS = [
-    os.path.basename(dirpath)
-    for dirpath, _, _ in os.walk("./mytest-tsfm")
-    if ".git" not in dirpath and "./mytest-tsfm" != dirpath and "finetuned" not in dirpath and "figures" not in dirpath
-]
-
+MODEL_IDS = ["ttm-1024-96-r1", "ttm-1024-96-r2", "ttm-1536-96-r2", "ttm-r1", "ttm-r2"]
 
 
 def min_context_length(model_id):