Skip to content

Commit 312e463

Browse files
chore: disabling progress bar in QdrantDocumentStore tests (#2797)
* disabling progress bar * formatting
1 parent 66b8109 commit 312e463

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

integrations/qdrant/tests/test_document_store.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def document_store(self) -> QdrantDocumentStore:
3131
return_embedding=True,
3232
wait_result_from_api=True,
3333
use_sparse_embeddings=False,
34+
progress_bar=False,
3435
)
3536

3637
def test_init_is_lazy(self):
@@ -146,7 +147,7 @@ def test_sparse_configuration(self):
146147
assert sparse_config[SPARSE_VECTORS_NAME].modifier == rest.Modifier.IDF
147148

148149
def test_query_hybrid(self, generate_sparse_embedding):
149-
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True)
150+
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True, progress_bar=False)
150151

151152
docs = []
152153
for i in range(20):
@@ -171,7 +172,7 @@ def test_query_hybrid(self, generate_sparse_embedding):
171172
assert document.embedding
172173

173174
def test_query_hybrid_with_group_by(self, generate_sparse_embedding):
174-
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True)
175+
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True, progress_bar=False)
175176

176177
docs = []
177178
for i in range(20):

integrations/qdrant/tests/test_document_store_async.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ def document_store(self) -> QdrantDocumentStore:
2727
return_embedding=True,
2828
wait_result_from_api=True,
2929
use_sparse_embeddings=False,
30+
progress_bar=False,
3031
)
3132

3233
@pytest.mark.asyncio
@@ -59,7 +60,7 @@ async def test_sparse_configuration_async(self):
5960

6061
@pytest.mark.asyncio
6162
async def test_query_hybrid_async(self, generate_sparse_embedding):
62-
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True)
63+
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True, progress_bar=False)
6364

6465
docs = []
6566
for i in range(20):
@@ -84,7 +85,7 @@ async def test_query_hybrid_async(self, generate_sparse_embedding):
8485

8586
@pytest.mark.asyncio
8687
async def test_query_hybrid_with_group_by_async(self, generate_sparse_embedding):
87-
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True)
88+
document_store = QdrantDocumentStore(location=":memory:", use_sparse_embeddings=True, progress_bar=False)
8889

8990
docs = []
9091
for i in range(20):

0 commit comments

Comments
 (0)