From 34389b848640abfb501e12707e1642fc1c1d1401 Mon Sep 17 00:00:00 2001 From: Chaitanya Lakhchaura <90706514+ZenithFlux@users.noreply.github.com> Date: Wed, 22 May 2024 12:28:02 +0530 Subject: [PATCH] fix: fixed `test_numpy_mean` frontend test --- .../test_numpy/test_statistics/test_averages_and_variances.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ivy_tests/test_ivy/test_frontends/test_numpy/test_statistics/test_averages_and_variances.py b/ivy_tests/test_ivy/test_frontends/test_numpy/test_statistics/test_averages_and_variances.py index c83ba971358b6..4405d2980ac80 100644 --- a/ivy_tests/test_ivy/test_frontends/test_numpy/test_statistics/test_averages_and_variances.py +++ b/ivy_tests/test_ivy/test_frontends/test_numpy/test_statistics/test_averages_and_variances.py @@ -222,7 +222,7 @@ def test_numpy_mean( on_device, keep_dims, ): - input_dtypes, x, axis = dtype_and_x + input_dtypes, x, axis, _, _ = dtype_and_x where, input_dtypes, test_flags = np_frontend_helpers.handle_where_and_array_bools( where=where, input_dtype=input_dtypes, @@ -239,6 +239,7 @@ def test_numpy_mean( axis=axis, dtype=dtype[0], out=None, + atol=0.0005, keepdims=keep_dims, where=where, )