From 0ba6c6aa27d56d78c9e4d85dd49c2c91c5e0a82f Mon Sep 17 00:00:00 2001 From: "a.khokhulin" Date: Mon, 25 Nov 2024 16:13:31 +0000 Subject: [PATCH] last line --- turbo_alignment/dataset/pair_preferences/pair_preference.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/turbo_alignment/dataset/pair_preferences/pair_preference.py b/turbo_alignment/dataset/pair_preferences/pair_preference.py index e163e4b..bc2498e 100755 --- 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 \ No newline at end of file + return new_instance