style: ci fixes

explodinggradients · Dec 13, 2024 · 4bc50f4 · 4bc50f4
1 parent 9fc57d2
commit 4bc50f4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/ragas/optimizers/genetic.py b/src/ragas/optimizers/genetic.py
@@ -519,9 +519,8 @@ def dict_to_str(dict: t.Dict[str, t.Any]) -> str:
                         ),
                         expected_output=dataset[idx]["prompts"][prompt_name][
                             "edited_output"
-                        ] or dataset[idx]["prompts"][prompt_name][
-                            "prompt_output"
-                        ],
+                        ]
+                        or dataset[idx]["prompts"][prompt_name]["prompt_output"],
                     )
                     for idx in indices
                 ]

diff --git a/tests/e2e/test_langchain_llm_attributes.py b/tests/e2e/test_langchain_llm_attributes.py
@@ -3,7 +3,7 @@
 from langchain_aws import ChatBedrock, ChatBedrockConverse
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_google_vertexai import ChatVertexAI
-from langchain_openai import AzureChatOpenAI, ChatOpenAI
+from langchain_openai import ChatOpenAI
 
 models = [
     ChatOpenAI(model="gpt-4o"),