linters fix

turbo-llm · Nov 23, 2024 · 6d0e565 · 6d0e565
1 parent a8d8f10
commit 6d0e565
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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