From ad96e994086c8da92467b307297cdfaafc65733f Mon Sep 17 00:00:00 2001
From: crawlingcub <86861129+crawlingcub@users.noreply.github.com>
Date: Fri, 1 Oct 2021 16:01:36 -0500
Subject: [PATCH] Fixing flaky tests

---
 tests/attr/test_shapley.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/attr/test_shapley.py b/tests/attr/test_shapley.py
index a4fe3f8e7e..d842082a24 100644
--- a/tests/attr/test_shapley.py
+++ b/tests/attr/test_shapley.py
@@ -371,7 +371,8 @@ def _multi_input_batch_scalar_shapley_assert(self, func: Callable) -> None:
             feature_mask=(mask1, mask2, mask3),
             perturbations_per_eval=(1,),
             target=None,
-            n_samples=800,
+            n_samples=3500,
+            delta=1.2,
         )
 
     def _shapley_test_assert(