easier to just use an include list at this point

ibm-granite · Jan 8, 2025 · f192485 · f192485
1 parent cf1aac1
commit f192485
Showing 1 changed file with 1 addition and 7 deletions.
diff --git 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):