Check if samples are the same

activeloopai · Aug 4, 2023 · a286259 · a286259
1 parent 7f48f19
commit a286259
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/deeplake/core/vectorstore/test_deeplake_vectorstore.py b/deeplake/core/vectorstore/test_deeplake_vectorstore.py
@@ -475,7 +475,9 @@ def test_search_quantitative(distance_metric, hub_cloud_dev_token):
 
     # use python implementation to search the data
     data_p = vector_store.search(
-        embedding=query_embedding, exec_option="python", distance_metric=distance_metric
+        embedding=query_embedding,
+        exec_option="tensor_db",
+        distance_metric=distance_metric,
     )
 
     # use indra implementation to search the data
@@ -486,6 +488,12 @@ def test_search_quantitative(distance_metric, hub_cloud_dev_token):
     )
 
     assert len(data_p["score"]) == len(data_ce["score"])
+
+    print(data_ce["id"])
+    print(data_p["id"])
+
+    assert data_ce["id"] == data_p["id"]
+
     assert all(
         [
             isclose(