Attempt to fix codestyle issues; probably fail one more time

explodinggradients · Feb 28, 2024 · f7500b7 · f7500b7
1 parent d42e93d
commit f7500b7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ragas/testset/generator.py b/src/ragas/testset/generator.py
@@ -97,6 +97,7 @@ def with_openai(
             generator_llm_model=generator_llm_model,
             embeddings_model=embeddings_model,
             critic_llm_model=critic_llm_model,
+            docstore=docstore,
             run_config=run_config,
         )
 
@@ -146,7 +147,7 @@ def _common_constructor(
         cls,
         chunk_size: int,
         generator_llm_model: LangchainLLMWrapper,
-        embeddings_model: LangchainLLMWrapper,
+        embeddings_model: BaseRagasEmbeddings,
         critic_llm_model: LangchainLLMWrapper,
         docstore: t.Optional[DocumentStore],
         run_config: t.Optional[RunConfig],