Black formatting added

m-star18 · Jan 7, 2021 · 5546bb9 · 5546bb9
1 parent 8114e83
commit 5546bb9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/benchmarks/benchmark_to_pytorch_from_cloud.py b/benchmarks/benchmark_to_pytorch_from_cloud.py
@@ -5,10 +5,9 @@
 from hub import Dataset
 from hub.schema import Tensor
 
+
 def benchmark():
-    schema = {
-        "image": Tensor((256, 256, 3), dtype="uint8")
-    }
+    schema = {"image": Tensor((256, 256, 3), dtype="uint8")}
     arr = (np.random.rand(256, 256, 3) * 100).astype("uint8")
     # ds = Dataset("s3://snark-test/superficial_dataset", mode="w", schema=schema, shape=(5000,))
     # for i in tqdm(range(len(ds))):
@@ -20,5 +19,6 @@ def benchmark():
     for i, b in enumerate(tqdm(dl)):
         pass
 
+
 if __name__ == "__main__":
-    benchmark()
+    benchmark()
diff --git a/hub/store/s3_file_system_replacement.py b/hub/store/s3_file_system_replacement.py
@@ -3,6 +3,7 @@
 from hub.store.s3_storage import S3Storage
 from zarr import MemoryStore
 
+
 class S3FileSystemReplacement(S3FileSystem):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)