Hub Dataset compute() added to TileDB benchmarks

activeloopai · Jan 28, 2021 · b986d17 · b986d17
1 parent 3deb9d0
commit b986d17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/benchmark_tiledb_hub.py b/benchmarks/benchmark_tiledb_hub.py
@@ -48,8 +48,8 @@ def time_hub(dataset, batch_size=1):
         t0 = time()
         for batch in range(ds.shape[0] // batch_size):
             x, y = (
-                ds[batch * batch_size : (batch + 1) * batch_size]["image"],
-                ds[batch * batch_size : (batch + 1) * batch_size]["label"],
+                ds[batch * batch_size : (batch + 1) * batch_size]["image"].compute(),
+                ds[batch * batch_size : (batch + 1) * batch_size]["label"].compute(),
             )
             counter += 1
             t1 = time()