fix: ivy.is_complex_dtype is crashing for bfloat16 in case no backend…

… is set. (#23978)
ivy-llc · Sep 18, 2023 · dca2ac4 · dca2ac4
1 parent cc0e4ea
commit dca2ac4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy_tests/test_ivy/helpers/function_testing.py b/ivy_tests/test_ivy/helpers/function_testing.py
@@ -542,7 +542,7 @@ def test_function(
         test_flags.test_gradients
         and not test_flags.instance_method
         and "bool" not in input_dtypes
-        and not any(ivy.is_complex_dtype(d) for d in input_dtypes)
+        and not any(d in ["complex64", "complex128"] for d in input_dtypes)
     ):
         if backend_to_test not in fw_list or not ivy.nested_argwhere(
             all_as_kwargs_np,