From cb0d344e316e8e65b948b2ec5db2f8619127cc13 Mon Sep 17 00:00:00 2001 From: Felix Hirwa Nshuti Date: Mon, 13 Feb 2023 18:15:23 +0530 Subject: [PATCH] cast to float dtypes for nansum (#10512) --- .../test_experimental/test_core/test_elementwise.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ivy_tests/test_ivy/test_functional/test_experimental/test_core/test_elementwise.py b/ivy_tests/test_ivy/test_functional/test_experimental/test_core/test_elementwise.py index 9061cbb9a695f..ac8708f80b627 100644 --- a/ivy_tests/test_ivy/test_functional/test_experimental/test_core/test_elementwise.py +++ b/ivy_tests/test_ivy/test_functional/test_experimental/test_core/test_elementwise.py @@ -477,10 +477,13 @@ def _get_castable_dtypes_values(draw, *, allow_nan=False): ) axis = draw(helpers.get_axis(shape=shape, force_int=True)) dtype1, values, dtype2 = draw( - helpers.get_castable_dtype(draw(available_dtypes), dtype[0], values[0]) + helpers.get_castable_dtype( + draw(helpers.get_dtypes("float")), dtype[0], values[0] + ) ) return [dtype1], [values], axis, dtype2 + # nansum @handle_test( fn_tree="functional.ivy.experimental.nansum",