Fix default value

aai-institute · Apr 12, 2024 · d691943 · d691943
1 parent 67a5b06
commit d691943
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pydvl/value/shapley/common.py b/src/pydvl/value/shapley/common.py
@@ -16,15 +16,15 @@
 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"]
 
 
 def compute_shapley_values(
     u: Utility,
     *,
-    done: StoppingCriterion,
+    done: StoppingCriterion = MaxChecks(None),
     mode: ShapleyMode = ShapleyMode.TruncatedMontecarlo,
     n_jobs: int = 1,
     seed: Optional[Seed] = None,