style

huggingface · Nov 18, 2022 · 532ae18 · 532ae18 · github-actions · Nov 18, 2022
1 parent 2e270dc
commit 532ae18
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 10 deletions.
diff --git a/src/datasets/arrow_dataset.py b/src/datasets/arrow_dataset.py
@@ -1314,9 +1314,7 @@ def save_to_disk(
                         shard_lengths[job_id], shard_sizes[job_id] = content
                     else:
                         pbar.update(content)
-        with fs.open(
-            path_join(dataset_path, config.DATASET_STATE_JSON_FILENAME), "w", encoding="utf-8"
-        ) as state_file:
+        with fs.open(path_join(dataset_path, config.DATASET_STATE_JSON_FILENAME), "w", encoding="utf-8") as state_file:
             json.dump(state, state_file, indent=2, sort_keys=True)
         with fs.open(
             path_join(dataset_path, config.DATASET_INFO_FILENAME), "w", encoding="utf-8"

diff --git a/tests/test_arrow_dataset.py b/tests/test_arrow_dataset.py
@@ -3560,7 +3560,7 @@ def test_dummy_dataset_serialize_fs(dataset, mock_fsspec, tmp_path_factory):
     dataset_path = "mock://my_dataset"
     storage_options = {
         "local_root_dir": tmp_path_factory.mktemp("test_dummy_dataset_serialize_fs"),
-        "auto_mkdir": True
+        "auto_mkdir": True,
     }
     dataset.save_to_disk(dataset_path, storage_options=storage_options)
     reloaded = dataset.load_from_disk(dataset_path, storage_options=storage_options)

diff --git a/tests/test_dataset_dict.py b/tests/test_dataset_dict.py
@@ -12,12 +12,7 @@
 from datasets.features import ClassLabel, Features, Sequence, Value
 from datasets.splits import NamedSplit
 
-from .utils import (
-    assert_arrow_memory_doesnt_increase,
-    assert_arrow_memory_increases,
-    require_tf,
-    require_torch,
-)
+from .utils import assert_arrow_memory_doesnt_increase, assert_arrow_memory_increases, require_tf, require_torch
 
 
 class DatasetDictTest(TestCase):