From 6d0e565b078db42eeada6edf1fa5e6dbb059a521 Mon Sep 17 00:00:00 2001 From: "a.khokhulin" Date: Sun, 24 Nov 2024 01:03:10 +0300 Subject: [PATCH] linters fix --- turbo_alignment/dataset/chat/chat.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/turbo_alignment/dataset/chat/chat.py b/turbo_alignment/dataset/chat/chat.py index c3a7656..2cd23e1 100755 --- a/turbo_alignment/dataset/chat/chat.py +++ b/turbo_alignment/dataset/chat/chat.py @@ -376,8 +376,7 @@ def get_slice(self, start: int, end: int) -> Self: new_instance.records = self.records[start:end] new_instance.original_records_map = { - record['id']: self.get_original_record_by_id(record['id']) - for record in dataset_records + record['id']: self.get_original_record_by_id(record['id']) for record in dataset_records } return new_instance