SampleBatch is always a list[SampleT]

Author: mdbenito

src/pydvl/valuation/types.py

@@ -15,7 +15,6 @@
     Any,
     Callable,
     Generator,
-    Iterable,
     Protocol,
     Type,
     TypeVar,
@@ -243,7 +242,7 @@ def __eq__(self, other: object) -> bool:
 
 SampleT = TypeVar("SampleT", bound=Sample)
 
-SampleBatch = Iterable[SampleT]
+SampleBatch = list[SampleT]
 SampleGenerator = Generator[SampleT, None, None]
 BatchGenerator = Generator[SampleBatch[SampleT], None, None]