use natural np for input_type

Signed-off-by: 11happy <[email protected]>
openvinotoolkit · Dec 30, 2024 · 5915c5d · 5915c5d
1 parent 08e1682
commit 5915c5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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