From 16edb11e488f1dcbe1e1202ec926c3ab318f5438 Mon Sep 17 00:00:00 2001
From: Nikola Obradovic <132568163+nobradovictt@users.noreply.github.com>
Date: Fri, 15 Nov 2024 15:09:30 +0100
Subject: [PATCH] [Fix CI] Fix TT-Explorer tests in Ci. (#1283)

---
 tools/explorer/test/run_tests.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tools/explorer/test/run_tests.py b/tools/explorer/test/run_tests.py
index 05a84dd1a..ceff14ae0 100644
--- a/tools/explorer/test/run_tests.py
+++ b/tools/explorer/test/run_tests.py
@@ -13,11 +13,11 @@
 PORT = 8002
 COMMAND_URL = "http://" + HOST + ":" + str(PORT) + "/apipost/v1/send_command"
 TEST_LOAD_MODEL_PATHS = [
-    "test/ttmlir/Dialect/TTNN/mnist_sharding.mlir",
+    "test/ttmlir/Dialect/TTNN/optimizer/mnist_sharding.mlir",
     "tools/explorer/test/models/*.mlir",
 ]
 TEST_EXECUTE_MODEL_PATHS = [
-    "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+    "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
 ]
 
 
@@ -97,14 +97,14 @@ def test_execute_model(model_path):
 
 def test_execute_mnist_l1_interleaved():
     execute_command(
-        "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+        "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
         {"optimizationPolicy": "L1 Interleaved"},
     )
 
 
 def test_execute_mnist_optimizer_disabled():
     execute_command(
-        "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+        "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
         {"optimizationPolicy": "Optimizer Disabled"},
     )