Fix path to test goldens.

nod-ai · Feb 6, 2024 · c20eba5 · c20eba5
1 parent 6a88016
commit c20eba5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/turbine_models/tests/stateless_llama_test.py b/models/turbine_models/tests/stateless_llama_test.py
@@ -65,7 +65,7 @@ def test_vmfb_comparison(self):
             target_triple="host",
         )
 
-        torch_str_cache_path = f"python/turbine_models/tests/vmfb_comparison_cached_torch_output_{precision}_{quantization}.txt"
+        torch_str_cache_path = f"models/turbine_models/tests/vmfb_comparison_cached_torch_output_{precision}_{quantization}.txt"
         # if cached, just read
         if os.path.exists(torch_str_cache_path):
             with open(torch_str_cache_path, "r") as f:
@@ -106,7 +106,7 @@ def test_streaming_vmfb_comparison(self):
             vmfb_path="streaming_llama.vmfb",
         )
 
-        torch_str_cache_path = f"python/turbine_models/tests/vmfb_comparison_cached_torch_output_{precision}_{quantization}.txt"
+        torch_str_cache_path = f"models/turbine_models/tests/vmfb_comparison_cached_torch_output_{precision}_{quantization}.txt"
         # if cached, just read
         if os.path.exists(torch_str_cache_path):
             with open(torch_str_cache_path, "r") as f: