From 166f06829859a1d78e5a0189a5e8f9837f64a5c0 Mon Sep 17 00:00:00 2001 From: Anes Benmerzoug Date: Wed, 13 Dec 2023 13:13:40 +0100 Subject: [PATCH] Fix import errors --- src/pydvl/utils/progress.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/pydvl/utils/progress.py b/src/pydvl/utils/progress.py index 7bd62aff4..03207de67 100644 --- a/src/pydvl/utils/progress.py +++ b/src/pydvl/utils/progress.py @@ -1,13 +1,11 @@ -from collections.abc import Iterator from itertools import cycle, takewhile -from typing import TYPE_CHECKING, Collection +from typing import TYPE_CHECKING, Collection, Iterator from tqdm.auto import tqdm -from pydvl.value.stopping import StoppingCriterion - if TYPE_CHECKING: from pydvl.value.result import ValuationResult + from pydvl.value.stopping import StoppingCriterion __all__ = ["repeat_indices"] @@ -15,7 +13,7 @@ def repeat_indices( indices: Collection[int], result: "ValuationResult", - done: StoppingCriterion, + done: "StoppingCriterion", **kwargs ) -> Iterator[int]: """Helper function to cycle indefinitely over a collection of indices