From f9beb6c111da95d5fc96df9e998b95a077d1c390 Mon Sep 17 00:00:00 2001 From: Dan Saattrup Nielsen Date: Tue, 30 Apr 2024 16:40:36 +0200 Subject: [PATCH] tests: Fix test config setup --- tests/conftest.py | 26 ++++++++++---------------- 1 file changed, 10 insertions(+), 16 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index fc8fc8b3..2f66bd4d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -31,27 +31,21 @@ def config() -> typing.Generator[DictConfig, None, None]: yield DictConfig( dict( dirs=dict(data="data", raw="raw", processed="processed", final="final"), - document_store=dict(name="jsonl", jsonl=dict(filename=file.name)), + document_store=dict(name="jsonl", filename=file.name), embedder=dict( name="e5", - e5=dict( - model_id="intfloat/multilingual-e5-small", - document_text_field="text", - ), - ), - embedding_store=dict( - name="numpy", numpy=dict(num_documents_to_retrieve=3) + model_id="intfloat/multilingual-e5-small", + document_text_field="text", ), + embedding_store=dict(name="numpy", num_documents_to_retrieve=3), generator=dict( name="openai", - openai=dict( - api_key_variable_name="OPENAI_API_KEY", - model="gpt-3.5-turbo", - temperature=0.0, - stream=False, - timeout=60, - max_tokens=128, - ), + api_key_variable_name="OPENAI_API_KEY", + model="gpt-3.5-turbo", + temperature=0.0, + stream=False, + timeout=60, + max_tokens=128, ), demo=dict( name="danish",