From b6b03d47f4feadc10f77ad0f9d53d122df49e971 Mon Sep 17 00:00:00 2001 From: Andrej Jakovljevic Date: Mon, 2 Dec 2024 11:26:56 +0000 Subject: [PATCH] Rebased to main --- 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 6ec79c7b..a2823082 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 = []