[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

tests/streaming/test_dataloader.py

@@ -11,12 +11,13 @@
 
 
 def compress(index):
-    return (index, index ** 2)
+    return (index, index**2)
 
 
 def another_compress(index):
     return (index, index * 2)
 
+
 class TestStatefulDataset:
     def __init__(self, size, step):
         self.size = size
@@ -213,7 +214,7 @@ def test_resume_single_dataset_dataloader_from_checkpoint(tmpdir):
             break
 
     # load the state dict
-    with open(os.path.join(tmpdir, "state_dict.json"), "r") as f:
+    with open(os.path.join(tmpdir, "state_dict.json")) as f:
         state_dict = json.load(f)
 
     # create a new dataloader
@@ -242,14 +243,13 @@ def test_resume_combined_dataset_dataloader_from_checkpoint(tmpdir):
     )
     optimize(
         fn=another_compress,
-        inputs=list(range(10,20)),
+        inputs=list(range(10, 20)),
         num_workers=2,
         output_dir=output_dir_2,
         chunk_size=3,
     )
 
-    ds = CombinedStreamingDataset(
-        [StreamingDataset(output_dir_1), StreamingDataset(output_dir_2)], seed=42)
+    ds = CombinedStreamingDataset([StreamingDataset(output_dir_1), StreamingDataset(output_dir_2)], seed=42)
 
     dataloader = StreamingDataLoader(ds, batch_size=2, num_workers=2, pin_memory=True)
 
@@ -261,12 +261,11 @@ def test_resume_combined_dataset_dataloader_from_checkpoint(tmpdir):
             break
 
     # load the state dict
-    with open(os.path.join(tmpdir, "state_dict.json"), "r") as f:
+    with open(os.path.join(tmpdir, "state_dict.json")) as f:
         state_dict = json.load(f)
 
     # create a new dataloader
-    ds = CombinedStreamingDataset(
-        [StreamingDataset(output_dir_1), StreamingDataset(output_dir_2)], seed=42)
+    ds = CombinedStreamingDataset([StreamingDataset(output_dir_1), StreamingDataset(output_dir_2)], seed=42)
     dataloader = StreamingDataLoader(ds, batch_size=2, num_workers=2, pin_memory=True)
     dataloader.load_state_dict(state_dict)