Fix formatting

adap · Aug 30, 2024 · d63ab02 · d63ab02
1 parent cb69493
commit d63ab02
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datasets/flwr_datasets/mock_utils_test.py b/datasets/flwr_datasets/mock_utils_test.py
@@ -428,8 +428,9 @@ def _load_mocked_dataset_dict_by_partial_download(
     subset_name: Optional[str] = None,
 ) -> DatasetDict:
     """Like _load_mocked_dataset_by_partial_download but for many splits."""
-    assert len(split_names) == len(skip_take_lists), "The split_names should be the"
-    "same length as the skip_take_lists."
+    assert len(split_names) == len(
+        skip_take_lists
+    ), "The split_names should be thesame length as the skip_take_lists."
     dataset_dict = {}
     for split_name, skip_take_list in zip(split_names, skip_take_lists):
         dataset_dict[split_name] = _load_mocked_dataset_by_partial_download(