[AL-2350] Indra returns polygons differently than python

deeplake/core/dataset/deeplake_query_tensor.py

@@ -68,7 +68,7 @@
         self, aslist=False, *args, **kwargs
     ) -> Union[np.ndarray, List[np.ndarray]]:
         r = self.indra_tensor.numpy(aslist=aslist)
-        if aslist or isinstance(r, np.ndarray):
+        if aslist or isinstance(r, (np.ndarray, list)):
             return r
         else:
             try:

deeplake/core/tests/test_deeplake_indra_dataset.py

@@ -10,6 +10,7 @@
 from deeplake.cli.auth import logout
 from deeplake.core.dataset.deeplake_query_dataset import DeepLakeQueryDataset
 import random
+import math
 import pytest
 
 
@@ -269,6 +270,45 @@
     assert deeplake_indra_ds[1].image.numpy()[98].shape == (20, 10, 3)
 
 
+@requires_libdeeplake
+def test_query_tensors_polygon_htype_consistency(local_auth_ds_generator):
+    ds = local_auth_ds_generator()
+    with ds:
+        ds.create_tensor(
+            "polygon",
+            dtype=np.float32,
+            htype="polygon",
+            sample_compression=None,
+        )
+        ds.create_tensor(
+            "labels",
+            dtype=np.uint16,
+            htype="generic",
+            sample_compression=None,
+        )
+        for i in range(10):
+            polygons = []
+            for j in range(i):
+                p = np.ndarray((3 + j, 2))
+                for k in range(3 + j):
+                    p[k] = [
+                        200 * (j % 3) + 150 * (math.sin(6.28 * k / (3 + j)) + 1) / 2,
+                        200 * (j / 3) + 150 * (math.cos(6.28 * k / (3 + j)) + 1) / 2,
+                    ]
+                polygons.append(p)
+            ds.labels.append(i)
+            ds.polygon.append(polygons)
+
+    view = ds.query("select *, labels+labels as new_tensor")
+    for i in range(len(ds)):
+        orig = ds.polygon[i].numpy()
+        new = view.polygon[i].numpy()
+
+        assert type(orig) == type(new)
+        for i, j in zip(orig, new):
+            assert np.all(i == j)
+
+
 @requires_libdeeplake
 def test_random_split(local_auth_ds_generator):
     deeplake_ds = local_auth_ds_generator()

deeplake/core/vectorstore/test_deeplake_vectorstore.py

@@ -721,7 +721,7 @@ def assert_updated_vector_store(
         ("local_auth_ds", None, "vector_store_row_ids", None, None),
         ("local_auth_ds", None, None, "vector_store_filter_udf", None),
         ("local_auth_ds", None, None, "vector_store_filters", None),
-        ("local_auth_ds", None, None, None, "vector_store_query"),
+        ("hub_cloud_ds", None, None, None, "vector_store_query"),
     ],
     indirect=True,
 )