From 5915c5d1b403f6b93242ae108a25fade0dcdd6c4 Mon Sep 17 00:00:00 2001 From: 11happy Date: Mon, 30 Dec 2024 22:53:38 +0530 Subject: [PATCH] use natural np for input_type Signed-off-by: 11happy --- tests/layer_tests/jax_tests/test_iota.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/layer_tests/jax_tests/test_iota.py b/tests/layer_tests/jax_tests/test_iota.py index 6f35bd623fb290..f9357fd904a475 100644 --- a/tests/layer_tests/jax_tests/test_iota.py +++ b/tests/layer_tests/jax_tests/test_iota.py @@ -26,7 +26,7 @@ def jax_iota(dtype, size): @pytest.mark.parametrize("input_shape", [1,2,3]) - @pytest.mark.parametrize("input_type", ["jnp.float32"]) + @pytest.mark.parametrize("input_type", ["np.float32"]) @pytest.mark.nightly @pytest.mark.precommit @pytest.mark.precommit_jax_fe