diff --git a/src/lighteval/data.py b/src/lighteval/data.py index bb0a3ded2..5211f745f 100644 --- a/src/lighteval/data.py +++ b/src/lighteval/data.py @@ -72,7 +72,6 @@ def __init__( self.num_dataset_splits, self.splits = self.init_split_limits(num_dataset_splits) self.split_start, self.split_end = self.splits[0] - self.split_end = self.splits[0][1] def init_split_limits(self, num_dataset_splits): if num_dataset_splits >= self.total_size: @@ -122,7 +121,6 @@ def get_split_start_end(self, split_id: int) -> tuple[int, int]: tuple: A tuple containing the start and end indices of the split. """ self.split_start, self.split_end = self.splits[split_id] - self.split_end = self.splits[split_id][1] return self.split_start, self.split_end def splits_start_end_iterator(self) -> tuple[int, int]: