update

turbo-llm · Nov 23, 2024 · a8d8f10 · a8d8f10
1 parent b38e9d9
commit a8d8f10
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/turbo_alignment/dataset/chat/chat.py b/turbo_alignment/dataset/chat/chat.py
@@ -376,7 +376,8 @@ 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