Update all files Issue: activeloopai#205

Gyanachand1 · Nov 17, 2020 · b964055 · b964055
1 parent 63612dd
commit b964055
Show file tree

Hide file tree

Showing 7 changed files with 58 additions and 35 deletions.
diff --git a/examples/basic.py b/examples/basic.py
@@ -2,6 +2,7 @@
 import numpy as np
 from hub import dev_mode
 
+
 def main():
     # Tag is set {Username}/{Dataset}
     tag = "davitb/basic11"
@@ -16,7 +17,6 @@ def main():
         },
     )
 
-
     # Upload Data
     ds["image"][:] = np.ones((4, 512, 512))
     ds["label"][:] = np.ones((4, 512, 512))
@@ -26,5 +26,6 @@ def main():
     ds = Dataset(tag)
     print(ds["image"][0].compute())
 
+
 if __name__ == "__main__":
     main()
diff --git a/examples/big_image.py b/examples/big_image.py
@@ -29,4 +29,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/eurosat.py b/examples/eurosat.py
@@ -1,12 +1,15 @@
 import hub
 import torch
 
-if __name__ == "__main__":
+
+def main():
     ds = hub.Dataset("eurosat/eurosat-rgb")
 
     # 26000 samples in dataset, accessing values
     print(ds["image"][10].numpy())
-    print(ds["label", 15].numpy())  # alternate way to access, by specifying both key and sample number at once
+    print(
+        ds["label", 15].numpy()
+    )  # alternate way to access, by specifying both key and sample number at once
     print(ds["filename", 20:22].numpy())  # accessing multiple elements at once
 
     # Splitting into train and test sets
@@ -31,11 +34,19 @@
     train_loader = torch.utils.data.DataLoader(train_pt_ds, batch_size=2)
 
     for batch in train_loader:
-        print(batch["label"], batch["image"])  # pytorch tensors don't support text labels such as filename
+        print(
+            batch["label"], batch["image"]
+        )  # pytorch tensors don't support text labels such as filename
         break
 
     test_pt_ds = test_ds.to_pytorch()
     test_loader = torch.utils.data.DataLoader(test_pt_ds, batch_size=2)
     for batch in test_loader:
-        print(batch["label"], batch["image"])  # pytorch tensors don't support text labels such as filename
+        print(
+            batch["label"], batch["image"]
+        )  # pytorch tensors don't support text labels such as filename
         break
+
+
+if __name__ == "__main__":
+    main()
diff --git a/examples/large_dataset_build.py b/examples/large_dataset_build.py
@@ -22,10 +22,9 @@ def create_large_dataset():
     for i in range(len(ds) // 10):
         ds["image", i * 10 : i * 10 + 10] = i * array
     ds.commit()
-    
+
     ds = hub.Dataset("./data/examples/large_dataset_build")
     print(ds.keys, ds["image"].shape, ds["image"].dtype)
-
 
     # Read the dataset
     with hub.Dataset("./data/examples/large_dataset_build") as ds:
@@ -34,4 +33,4 @@ def create_large_dataset():
 
 
 if __name__ == "__main__":
-    create_large_dataset()
+    create_large_dataset()
diff --git a/examples/load_tf.py b/examples/load_tf.py
@@ -1,18 +1,24 @@
 from hub import Dataset, features
 
-# Create dataset
-ds = Dataset(
-    "./data/example/pytorch",
-    shape=(64,),
-    schema={
-        "image": features.Tensor((512, 512), dtype="float"),
-        "label": features.Tensor((512, 512), dtype="float"),
-    },
-)
 
-# tansform into Tensorflow dataset
-ds = ds.to_tensorflow().batch(8)
+def main():
+    # Create dataset
+    ds = Dataset(
+        "./data/example/pytorch",
+        shape=(64,),
+        schema={
+            "image": features.Tensor((512, 512), dtype="float"),
+            "label": features.Tensor((512, 512), dtype="float"),
+        },
+    )
 
-# Iterate over the data
-for batch in ds:
-    print(batch["image"], batch["label"])
+    # tansform into Tensorflow dataset
+    ds = ds.to_tensorflow().batch(8)
+
+    # Iterate over the data
+    for batch in ds:
+        print(batch["image"], batch["label"])
+
+
+if __name__ == "__main__":
+    main()
diff --git a/examples/mnist_upload_speed_bechmark.py b/examples/mnist_upload_speed_bechmark.py
@@ -33,4 +33,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/new_api_intro.py b/examples/new_api_intro.py
@@ -3,17 +3,23 @@
 from hub import Dataset
 from hub.features import ClassLabel, Image
 
-schema = {
-    "image": Image((28, 28)),
-    "label": ClassLabel(num_classes=10),
-}
 
-ds = Dataset("./data/examples/new_api_intro", shape=(1000,), schema=schema)
+def main():
+    schema = {
+        "image": Image((28, 28)),
+        "label": ClassLabel(num_classes=10),
+    }
 
-for i in range(len(ds)):
-    ds["image", i] = np.ones((28, 28), dtype="uint8")
-    ds["label", i] = 3
+    ds = Dataset("./data/examples/new_api_intro", shape=(1000,), schema=schema)
 
-print(ds["image", 5].numpy())
-print(ds["label", 100:110].numpy())
-ds.commit()
+    for i in range(len(ds)):
+        ds["image", i] = np.ones((28, 28), dtype="uint8")
+        ds["label", i] = 3
+
+    print(ds["image", 5].numpy())
+    print(ds["label", 100:110].numpy())
+    ds.commit()
+
+
+if __name__ == "__main__":
+    main()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -29,4 +29,4 @@ def main():


		if __name__ == "__main__":
		main()
		main()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -33,4 +33,4 @@ def main():


		if __name__ == "__main__":
		main()
		main()