Skip to content

Commit

Permalink
Fix import errors
Browse files Browse the repository at this point in the history
  • Loading branch information
AnesBenmerzoug committed Dec 13, 2023
1 parent 8a696ee commit 166f068
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions src/pydvl/utils/progress.py
Original file line number Diff line number Diff line change
@@ -1,21 +1,19 @@
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"]


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
Expand Down

0 comments on commit 166f068

Please sign in to comment.