fix TypeError: argument of type 'LazyStrategy' is not iterable (#6129)

ivy-llc · Oct 25, 2022 · a9074d7 · a9074d7
1 parent 643d888
commit a9074d7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ivy_tests/test_ivy/test_functional/test_core/test_general.py b/ivy_tests/test_ivy/test_functional/test_core/test_general.py
@@ -1189,12 +1189,14 @@ def test_einops_rearrange(
             ("b c (h1 h2) (w1 w2) -> b c h1 w1", {"h2": 2, "w2": 2}),
         ]
     ),
+    floattypes=helpers.get_dtypes("float"),
     reduction=st.sampled_from(["min", "max", "sum", "mean", "prod"]),
     num_positional_args=helpers.num_positional_args(fn_name="einops_reduce"),
 )
 def test_einops_reduce(
     dtype_x,
     pattern_and_axes_lengths,
+    floattypes,
     reduction,
     with_out,
     as_variable,
@@ -1206,7 +1208,7 @@ def test_einops_reduce(
 ):
     pattern, axes_lengths = pattern_and_axes_lengths
     dtype, x = dtype_x
-    if (reduction in ["mean", "prod"]) and (dtype not in helpers.get_dtypes("float")):
+    if (reduction in ["mean", "prod"]) and (dtype not in floattypes):
         dtype = ["float32"]
     helpers.test_function(
         input_dtypes=dtype,