Apply suggestions from code review

huggingface · Jul 10, 2024 · 5db69e8 · 5db69e8
1 parent 8a1814b
commit 5db69e8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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]: