Add additional variable to pass mypy

aai-institute · Sep 2, 2024 · b236c86 · b236c86
1 parent 784a40e
commit b236c86
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/pydvl/valuation/samplers/powerset.py b/src/pydvl/valuation/samplers/powerset.py
@@ -306,10 +306,8 @@ def _generate(self, indices: IndexSetT) -> SampleGenerator:
             for subset in powerset(
                 complement(indices, [idx] if idx is not None else [])
             ):
-                yield Sample(
-                    idx,
-                    np.asarray(cast(Collection[IndexT], subset), dtype=indices.dtype),
-                )
+                typed_subset: Collection[IndexT] = subset
+                yield Sample(idx, np.asarray(typed_subset, dtype=indices.dtype))
 
     def sample_limit(self, indices: IndexSetT) -> int | None:
         len_outer = self._index_iteration.length(indices)