diff --git a/tests/attr/test_shapley.py b/tests/attr/test_shapley.py index a4e57a5d57..6d894fbec5 100644 --- a/tests/attr/test_shapley.py +++ b/tests/attr/test_shapley.py @@ -349,6 +349,7 @@ def _single_int_input_multi_sample_batch_scalar_shapley_assert( feature_mask=mask, perturbations_per_eval=(1,), target=None, + n_samples=2500, ) def _multi_input_batch_scalar_shapley_assert(self, func: Callable) -> None: