Test

Author: Nikos Papailiou

apis/python/src/tiledb/vector_search/index.py

@@ -217,8 +217,8 @@ def distributed_query(
                 parts_uri=self.parts_db_uri,
                 ids_uri=self.ids_uri,
                 query_vectors=targets_m,
-                active_partitions=active_partitions[part:part_end],
-                active_queries=active_queries[part:part_end],
+                active_partitions=np.array(active_partitions)[part:part_end],
+                active_queries=np.array(active_queries[part:part_end]),
                 indices=self._index,
                 k_nn=k,
                 ctx=self.ctx,