Ignore some tqdm related type hints

aai-institute · Dec 11, 2023 · 27a85d1 · 27a85d1
1 parent d1aa6ad
commit 27a85d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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}",