Simplify

Author: mdbenito

src/pydvl/valuation/scorers/utils.py

@@ -37,7 +37,7 @@ def compose_score(
         The composite [SupervisedScorer][pydvl.valuation.scorers.SupervisedScorer].
     """
 
-    class CompositeSupervisedScorer(SupervisedScorer[SupervisedModelT, ArrayT]):
+    class CompositeSupervisedScorer(SupervisedScorer):
         def __call__(self, model: SupervisedModelT) -> float:
             raw = super().__call__(model)
             return transformation(raw)

src/pydvl/valuation/utility/classwise.py

@@ -6,14 +6,11 @@
 
 from __future__ import annotations
 
-from typing import cast
-
 import numpy as np
 
-from pydvl.utils.array import Array
 from pydvl.utils.caching import CacheBackend, CachedFuncConfig
 from pydvl.valuation.scorers.classwise import ClasswiseSupervisedScorer
-from pydvl.valuation.types import ClasswiseSample, IndexT, SupervisedModel
+from pydvl.valuation.types import ClasswiseSample, SupervisedModel
 from pydvl.valuation.utility import ModelUtility
 
 __all__ = ["ClasswiseModelUtility"]
@@ -74,9 +71,6 @@ def _utility(self, sample: ClasswiseSample) -> float:
         #   - combine the in-class and out-of-class subsets
         self.scorer.label = sample.label
         new_sample = sample.with_subset(
-            cast(
-                Array[IndexT],
-                np.union1d(sample.subset, sample.ooc_subset).astype(np.int_),
-            )
+            np.union1d(sample.subset, sample.ooc_subset).astype(np.int_),
         )
         return super()._utility(new_sample)