#4003: use reports_path instead of tmp_path

tenstorrent · Apr 3, 2024 · 2e240d3 · 2e240d3
1 parent 065ebb2
commit 2e240d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ttnn/ttnn/model_preprocessing.py b/ttnn/ttnn/model_preprocessing.py
@@ -427,7 +427,7 @@ def infer_ttnn_module_args(*, model, run_model, device):
     with trace():
         output = run_model(model)
 
-    visualize(output, file_name=ttnn.CONFIG.tmp_dir / "model_graph.svg")
+    visualize(output, file_name=ttnn.CONFIG.reports_path / "model_graph.svg")
 
     def _infer_ttnn_module_args(graph):
         ttnn_module_args = {}