fix: fixed test_numpy_mean frontend test

ivy-llc · May 22, 2024 · 34389b8 · 34389b8
1 parent aa64649
commit 34389b8
Showing 1 changed file with 2 additions and 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
@@ -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,
     )