Skip to content

Commit efea062

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
integration test fix for lancedb
1 parent 2f247da commit efea062

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/integration/vector_stores/test_lancedb.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ def test_vector_store_operations():
2020
try:
2121
vector_store = LanceDBVectorStore(
2222
vector_store_schema_config=VectorStoreSchemaConfig(
23-
index_name="test_collection"
23+
index_name="test_collection",
24+
vector_size=5
2425
)
2526
)
2627
vector_store.connect(db_uri=temp_dir)
@@ -98,7 +99,8 @@ def test_empty_collection():
9899
try:
99100
vector_store = LanceDBVectorStore(
100101
vector_store_schema_config=VectorStoreSchemaConfig(
101-
index_name="empty_collection"
102+
index_name="empty_collection",
103+
vector_size=5
102104
)
103105
)
104106
vector_store.connect(db_uri=temp_dir)
@@ -142,7 +144,8 @@ def test_filter_search():
142144
try:
143145
vector_store = LanceDBVectorStore(
144146
vector_store_schema_config=VectorStoreSchemaConfig(
145-
index_name="filter_collection"
147+
index_name="filter_collection",
148+
vector_size=5
146149
)
147150
)
148151

0 commit comments

Comments
 (0)