Merge pull request #2511 from activeloopai/fy_opt_view

[AL-2368] Optimization for saving shallow views
activeloopai · Aug 1, 2023 · 95eec4e · 95eec4e
2 parents bc5f7df + 66b4690
commit 95eec4e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/deeplake/core/dataset/dataset.py b/deeplake/core/dataset/dataset.py
@@ -3273,7 +3273,13 @@ def _write_vds(
                         create_shape_tensor=False,
                         create_id_tensor=False,
                         create_sample_info_tensor=False,
-                    ).extend(list(self.index.values[0].indices(self.num_samples)))
+                    ).extend(
+                        np.array(
+                            tuple(self.index.values[0].indices(self.num_samples)),
+                            dtype="uint64",
+                        ),
+                        progressbar=True,
+                    )
                     info["first-index-subscriptable"] = self.index.subscriptable_at(0)
                     if len(self.index) > 1:
                         info["sub-sample-index"] = Index(