Skip to content

Commit dc00cb8

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
uv fix for format
1 parent efea062 commit dc00cb8

File tree

2 files changed

+12
-9
lines changed

2 files changed

+12
-9
lines changed

tests/integration/vector_stores/test_factory.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,9 @@ def test_create_lancedb_vector_store():
2222
}
2323
vector_store = VectorStoreFactory.create_vector_store(
2424
vector_store_type=VectorStoreType.LanceDB.value,
25-
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
25+
vector_store_schema_config=VectorStoreSchemaConfig(
26+
index_name="test_collection"
27+
),
2628
kwargs=kwargs,
2729
)
2830
assert isinstance(vector_store, LanceDBVectorStore)
@@ -37,7 +39,9 @@ def test_create_azure_ai_search_vector_store():
3739
}
3840
vector_store = VectorStoreFactory.create_vector_store(
3941
vector_store_type=VectorStoreType.AzureAISearch.value,
40-
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
42+
vector_store_schema_config=VectorStoreSchemaConfig(
43+
index_name="test_collection"
44+
),
4145
kwargs=kwargs,
4246
)
4347
assert isinstance(vector_store, AzureAISearchVectorStore)
@@ -52,7 +56,9 @@ def test_create_cosmosdb_vector_store():
5256

5357
vector_store = VectorStoreFactory.create_vector_store(
5458
vector_store_type=VectorStoreType.CosmosDB.value,
55-
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
59+
vector_store_schema_config=VectorStoreSchemaConfig(
60+
index_name="test_collection"
61+
),
5662
kwargs=kwargs,
5763
)
5864

tests/integration/vector_stores/test_lancedb.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,7 @@ def test_vector_store_operations():
2020
try:
2121
vector_store = LanceDBVectorStore(
2222
vector_store_schema_config=VectorStoreSchemaConfig(
23-
index_name="test_collection",
24-
vector_size=5
23+
index_name="test_collection", vector_size=5
2524
)
2625
)
2726
vector_store.connect(db_uri=temp_dir)
@@ -99,8 +98,7 @@ def test_empty_collection():
9998
try:
10099
vector_store = LanceDBVectorStore(
101100
vector_store_schema_config=VectorStoreSchemaConfig(
102-
index_name="empty_collection",
103-
vector_size=5
101+
index_name="empty_collection", vector_size=5
104102
)
105103
)
106104
vector_store.connect(db_uri=temp_dir)
@@ -144,8 +142,7 @@ def test_filter_search():
144142
try:
145143
vector_store = LanceDBVectorStore(
146144
vector_store_schema_config=VectorStoreSchemaConfig(
147-
index_name="filter_collection",
148-
vector_size=5
145+
index_name="filter_collection", vector_size=5
149146
)
150147
)
151148

0 commit comments

Comments
 (0)