fix failing tests in JAX frontend related to changes in commit f84bea1

rgrupesh · Feb 22, 2023 · e95f73f · e95f73f
1 parent ece59ec
commit e95f73f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/ivy_tests/test_ivy/test_frontends/test_jax/test_func_wrapper.py b/ivy_tests/test_ivy/test_frontends/test_jax/test_func_wrapper.py
@@ -21,7 +21,9 @@ def _fn(x, check_default=False):
 
 
 @given(
-    dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("valid")),
+    dtype_and_x=helpers.dtype_and_values(
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False)
+    ),
 )
 def test_inputs_to_ivy_arrays(dtype_and_x):
     x_dtype, x = dtype_and_x
@@ -49,7 +51,9 @@ def test_inputs_to_ivy_arrays(dtype_and_x):
 
 
 @given(
-    dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("valid")),
+    dtype_and_x=helpers.dtype_and_values(
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False)
+    ),
 )
 def test_outputs_to_frontend_arrays(dtype_and_x):
     x_dtype, x = dtype_and_x
@@ -65,7 +69,9 @@ def test_outputs_to_frontend_arrays(dtype_and_x):
 
 
 @given(
-    dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("valid")),
+    dtype_and_x=helpers.dtype_and_values(
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False)
+    ),
 )
 def test_to_ivy_arrays_and_back(dtype_and_x):
     x_dtype, x = dtype_and_x

diff --git a/ivy_tests/test_ivy/test_frontends/test_jax/test_jax_devicearray.py b/ivy_tests/test_ivy/test_frontends/test_jax/test_jax_devicearray.py
@@ -13,7 +13,9 @@
 
 
 @given(
-    dtype_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("valid")),
+    dtype_x=helpers.dtype_and_values(
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False)
+    ),
 )
 def test_jax_devicearray_property_ivy_array(
     dtype_x,
@@ -30,7 +32,9 @@ def test_jax_devicearray_property_ivy_array(
 
 
 @given(
-    dtype_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("valid")),
+    dtype_x=helpers.dtype_and_values(
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False)
+    ),
 )
 def test_jax_devicearray_property_dtype(
     dtype_x,
@@ -42,7 +46,7 @@ def test_jax_devicearray_property_dtype(
 
 @given(
     dtype_x_shape=helpers.dtype_and_values(
-        available_dtypes=helpers.get_dtypes("valid"),
+        available_dtypes=helpers.get_dtypes("valid", prune_function=False),
         ret_shape=True,
     ),
 )
@@ -58,7 +62,7 @@ def test_jax_devicearray_property_shape(
 def _at_helper(draw):
     _, data, shape = draw(
         helpers.dtype_and_values(
-            available_dtypes=helpers.get_dtypes("valid"),
+            available_dtypes=helpers.get_dtypes("valid", prune_function=False),
             num_arrays=2,
             shared_dtype=True,
             min_num_dims=1,