From d770c943987cf251f5f239249d55fad76dc5ba34 Mon Sep 17 00:00:00 2001 From: Andrej Jakovljevic Date: Mon, 2 Dec 2024 13:10:17 +0100 Subject: [PATCH] Rebased to main (#91) --- tests/TTIR/test_basic_ops.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/TTIR/test_basic_ops.py b/tests/TTIR/test_basic_ops.py index 6ec79c7..a282308 100644 --- a/tests/TTIR/test_basic_ops.py +++ b/tests/TTIR/test_basic_ops.py @@ -233,11 +233,12 @@ def module_transpose(a): verify_module(module_transpose, input_shapes) -def test_scalar_type(): +@pytest.mark.parametrize("input_shapes", [[(3, 3)]]) +def test_scalar_type(input_shapes): def module_scalar_type(a): return a.shape[0] - verify_module(module_scalar_type, [(3, 3)]) + verify_module(module_scalar_type, input_shapes) dim0_cases = []