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,