Ignore type checking for lines that adhere to official langchain doc

explodinggradients · Feb 28, 2024 · a7ab74b · a7ab74b
1 parent f7500b7
commit a7ab74b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ragas/testset/generator.py b/src/ragas/testset/generator.py
@@ -119,13 +119,13 @@ def with_google(
             HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
         }
         generator_llm_model = LangchainLLMWrapper(
-            ChatGoogleGenerativeAI(
+            ChatGoogleGenerativeAI(  # type: ignore
                 model=generator_llm,
                 safety_settings=safety_blocknone,
             )
         )
         critic_llm_model = LangchainLLMWrapper(
-            ChatGoogleGenerativeAI(
+            ChatGoogleGenerativeAI(  # type: ignore
                 model=critic_llm,
                 safety_settings=safety_blocknone,
             )