rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-04 10:52:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: can't infer the vector column for multivector (#2026)

Browse Source 
Signed-off-by: BubbleCal <bubble-cal@outlook.com>
This commit is contained in:
BubbleCal
2025-01-16 14:08:04 +08:00
committed by GitHub
parent 5c759505b8
commit 493f9ce467
3 changed files with 28 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/python/tests/docs/test_multivector.py
View File

@@ -1,4 +1,5 @@
import shutil
from lancedb.index import IvfPq
import pytest
# --8<-- [start:imports]
@@ -28,6 +29,9 @@ def test_multivector():
]
tbl = db.create_table("my_table", data=data, schema=schema)
# only cosine similarity is supported for multi-vectors
tbl.create_index(metric="cosine")
# query with single vector
query = np.random.random(256)
tbl.search(query).to_arrow()
@@ -59,6 +63,9 @@ async def test_multivector_async():
]
tbl = await db.create_table("my_table", data=data, schema=schema)
# only cosine similarity is supported for multi-vectors
await tbl.create_index(column="vector", config=IvfPq(distance_type="cosine"))
# query with single vector
query = np.random.random(256)
await tbl.query().nearest_to(query).to_arrow()