From 27a85d14d4943a01585921d2058740cdb5871ed2 Mon Sep 17 00:00:00 2001 From: Anes Benmerzoug Date: Mon, 11 Dec 2023 09:41:34 +0100 Subject: [PATCH] Ignore some tqdm related type hints --- src/pydvl/value/least_core/naive.py | 4 ++-- src/pydvl/value/shapley/naive.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/pydvl/value/least_core/naive.py b/src/pydvl/value/least_core/naive.py index 9467292b3..f97021678 100644 --- a/src/pydvl/value/least_core/naive.py +++ b/src/pydvl/value/least_core/naive.py @@ -104,10 +104,10 @@ def lc_prepare_problem(u: Utility, progress: bool = False) -> LeastCoreProblem: logger.debug("Iterating over all subsets") utility_values = np.zeros(powerset_size) - for i, subset in enumerate( + for i, subset in enumerate( # type: ignore tqdm( powerset(u.data.indices), - display=progress, + disable=not progress, total=powerset_size - 1, position=0, ) diff --git a/src/pydvl/value/shapley/naive.py b/src/pydvl/value/shapley/naive.py index bd69432eb..031925681 100644 --- a/src/pydvl/value/shapley/naive.py +++ b/src/pydvl/value/shapley/naive.py @@ -78,7 +78,7 @@ def _combinatorial_exact_shapley( subset: NDArray[np.int_] = np.setxor1d( u.data.indices, [i], assume_unique=True ).astype(np.int_) - for s in tqdm( + for s in tqdm( # type: ignore powerset(subset), disable=not progress, desc=f"Index {i}",