last line

turbo-llm · Nov 25, 2024 · 0ba6c6a · 0ba6c6a
1 parent d8d811c
commit 0ba6c6a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/turbo_alignment/dataset/pair_preferences/pair_preference.py b/turbo_alignment/dataset/pair_preferences/pair_preference.py
@@ -43,6 +43,7 @@ def __init__(
             read=False,
         )
         super().__init__(source=source, settings=settings, tokenizer=tokenizer)
+        self.settings: PairPreferenceDatasetSettings = settings
 
         if read:
             self._read()
@@ -124,4 +125,4 @@ def get_slice(self, start: int, end: int) -> Self:
             record['id']: self.get_original_record_by_id(record['id']) for record in dataset_records
         }
 
-        return new_instance
+        return new_instance