#9049: chage shape parameter from tuple to list

tenstorrent · Jun 4, 2024 · e00ec32 · e00ec32
1 parent c6a1d9a
commit e00ec32
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/tt_eager/python_api_testing/unit_testing/misc/test_moreh_sgd.py b/tests/tt_eager/python_api_testing/unit_testing/misc/test_moreh_sgd.py
@@ -56,10 +56,10 @@ def create_tt_tensor(tensor, device):
 
 @pytest.mark.parametrize(
     "shape",
-    (
-        (32, 32),  # single
-        (12, 6, 64, 64),  # multiple tiles
-    ),
+    [
+        [32, 32],  # single
+        [12, 6, 64, 64],  # multiple tiles
+    ],
 )
 @pytest.mark.parametrize("lr", [3.0])
 @pytest.mark.parametrize("momentum", [0.0, 7.7])
@@ -195,7 +195,7 @@ def forward(self, x):
 
 @pytest.mark.parametrize(
     "shape",
-    ((32, 32),),  # single
+    [[32, 32]],  # single
 )
 @pytest.mark.parametrize("lr", [3.0])
 @pytest.mark.parametrize("momentum", [7.7])