fix style

Niccolo-Ajroldi · Niccolo-Ajroldi · commit 46b645b2ac4a · 2025-03-20T11:38:40.000+01:00
diff --git a/dataset/dataset_setup.py b/dataset/dataset_setup.py
@@ -797,7 +797,8 @@ def concat_chunck(examples: Dict[str, List[Any]]) -> Dict[str, List[Any]]:
         k: [
             t[i:i + max_seq_length]
             for i in range(0, total_length, max_seq_length)
-        ] for k, t in concatenated_examples.items()
+        ] for k,
+        t in concatenated_examples.items()
     }
     return result
 

Original file line number	Diff line number	Diff line change
`@@ -797,7 +797,8 @@ def concat_chunck(examples: Dict[str, List[Any]]) -> Dict[str, List[Any]]:`
`797`	`797`	`k: [`
`798`	`798`	`t[i:i + max_seq_length]`
`799`	`799`	`for i in range(0, total_length, max_seq_length)`
`800`		`- ] for k, t in concatenated_examples.items()`
	`800`	`+ ] for k,`
	`801`	`+ t in concatenated_examples.items()`
`801`	`802`	`}`
`802`	`803`	`return result`
`803`	`804`