From d0489da4017d45bd9ec650854c7da24b7f71eb17 Mon Sep 17 00:00:00 2001 From: egolearner Date: Tue, 12 May 2026 19:04:33 +0800 Subject: [PATCH] ut: fix some uts (#394) --- python/tests/test_hnsw_contiguous_memory.py | 6 +++--- python/tests/test_vamana.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/python/tests/test_hnsw_contiguous_memory.py b/python/tests/test_hnsw_contiguous_memory.py index cb23a01..380a0f9 100644 --- a/python/tests/test_hnsw_contiguous_memory.py +++ b/python/tests/test_hnsw_contiguous_memory.py @@ -46,7 +46,7 @@ from zvec import ( HnswIndexParam, HnswQueryParam, InvertIndexParam, - VectorQuery, + Query, VectorSchema, ) from zvec.typing import DataType, IndexType, MetricType, QuantizeType @@ -115,12 +115,12 @@ def _generate_docs(rng: np.random.Generator, num: int = NUM_DOCS) -> list[Doc]: def _assert_query_matches(coll: Collection, query_vec: list[float]) -> list[str]: """Run a top-k vector query and return the returned ids in order.""" - vector_query = VectorQuery( + vector_query = Query( field_name="dense", vector=query_vec, param=HnswQueryParam(ef=128), ) - hits = coll.query(vectors=vector_query, topk=TOPK) + hits = coll.query(vector_query, topk=TOPK) # Expect a single result group for the single vector query. assert hits is not None, "query returned None" assert len(hits) >= 1, f"expected at least one hit, got {hits!r}" diff --git a/python/tests/test_vamana.py b/python/tests/test_vamana.py index 359a9d8..50d6830 100644 --- a/python/tests/test_vamana.py +++ b/python/tests/test_vamana.py @@ -46,7 +46,7 @@ from zvec import ( InvertIndexParam, VamanaIndexParam, VamanaQueryParam, - VectorQuery, + Query, VectorSchema, ) from zvec.typing import DataType, IndexType, MetricType, QuantizeType @@ -125,12 +125,12 @@ def _query_topk( coll: Collection, query_vec: list[float], *, ef_search: int = 64 ) -> list[str]: """Run a top-k vector query and return the returned ids in order.""" - vector_query = VectorQuery( + vector_query = Query( field_name="dense", vector=query_vec, param=VamanaQueryParam(ef_search=ef_search), ) - hits = coll.query(vectors=vector_query, topk=TOPK) + hits = coll.query(vector_query, topk=TOPK) assert hits is not None, "query returned None" assert len(hits) >= 1, f"expected at least one hit, got {hits!r}" return [doc.id for doc in hits]