Change type parameter of our sampler (#268)

This is for aligning with pytorch
wearepal · Oct 9, 2023 · 2140e16 · 2140e16
1 parent 4152841
commit 2140e16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ranzen/torch/data.py b/ranzen/torch/data.py
@@ -264,7 +264,7 @@ class TrainingMode(Enum):
     """step-based training"""
 
 
-class BatchSamplerBase(Sampler[Sequence[int]]):
+class BatchSamplerBase(Sampler[List[int]]):
     def __init__(self, epoch_length: int | None = None) -> None:
         self.epoch_length: Final[int | None] = epoch_length