Fix types

Author: mdbenito

src/pydvl/reporting/plots.py

@@ -104,7 +104,7 @@ def plot_ci_array(
     means = np.mean(data, axis=0)
     variances = np.var(data, axis=0, ddof=1)
 
-    dummy: ValuationResult[np.int_, str] = ValuationResult(
+    dummy = ValuationResult[np.int_, np.object_](
         algorithm="dummy",
         values=means,
         variances=variances,

src/pydvl/value/semivalues.py

@@ -92,7 +92,7 @@
 import warnings
 from enum import Enum
 from itertools import islice
-from typing import Collection, List, Optional, Protocol, Tuple, Type, TypeVar, cast
+from typing import Iterable, List, Optional, Protocol, Tuple, Type, cast
 
 import scipy as sp
 from deprecate import deprecated
@@ -143,7 +143,7 @@ def __call__(self, n: int, k: int) -> float:
 
 
 def _marginal(
-    u: Utility, coefficient: SVCoefficient, samples: Collection[SampleT]
+    u: Utility, coefficient: SVCoefficient, samples: Iterable[SampleT]
 ) -> Tuple[MarginalT, ...]:
     """Computation of marginal utility. This is a helper function for
     [compute_generic_semivalues][pydvl.value.semivalues.compute_generic_semivalues].
@@ -278,7 +278,7 @@ def compute_generic_semivalues(
                         raise StopIteration
 
                     # Filter out samples for indices that have already converged
-                    filtered_samples = samples
+                    filtered_samples: Iterable = samples
                     if skip_converged and len(done.converged) > 0:
                         # t[0] is the index for the sample
                         filtered_samples = filter(