diff --git a/deeplake/core/vectorstore/test_deeplake_vectorstore.py b/deeplake/core/vectorstore/test_deeplake_vectorstore.py index 54eb44d359..c3bdbc24f2 100644 --- 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="python", + 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(