From d691943f7d674c2151b55acdd276be9c321a30c1 Mon Sep 17 00:00:00 2001
From: Miguel de Benito Delgado <m.debenito@appliedai.de>
Date: Fri, 12 Apr 2024 17:46:22 +0200
Subject: [PATCH] Fix default value

---
 src/pydvl/value/shapley/common.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/pydvl/value/shapley/common.py b/src/pydvl/value/shapley/common.py
index c9a767aa7..2f07b5072 100644
--- a/src/pydvl/value/shapley/common.py
+++ b/src/pydvl/value/shapley/common.py
@@ -16,7 +16,7 @@
 from pydvl.value.shapley.owen import OwenAlgorithm, owen_sampling_shapley
 from pydvl.value.shapley.truncated import NoTruncation
 from pydvl.value.shapley.types import ShapleyMode
-from pydvl.value.stopping import StoppingCriterion
+from pydvl.value.stopping import MaxChecks, StoppingCriterion
 
 __all__ = ["compute_shapley_values"]
 
@@ -24,7 +24,7 @@
 def compute_shapley_values(
     u: Utility,
     *,
-    done: StoppingCriterion,
+    done: StoppingCriterion = MaxChecks(None),
     mode: ShapleyMode = ShapleyMode.TruncatedMontecarlo,
     n_jobs: int = 1,
     seed: Optional[Seed] = None,