From 72db52b5d4bfef17f869eef137ac9ef047d8ff4a Mon Sep 17 00:00:00 2001 From: vishah02 Date: Wed, 11 Dec 2024 13:48:20 -0500 Subject: [PATCH] linting fixed in some of the tests --- libs/genai/tests/integration_tests/test_chat_models.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/libs/genai/tests/integration_tests/test_chat_models.py b/libs/genai/tests/integration_tests/test_chat_models.py index 304c9a2a..3215e64b 100644 --- a/libs/genai/tests/integration_tests/test_chat_models.py +++ b/libs/genai/tests/integration_tests/test_chat_models.py @@ -2,10 +2,9 @@ import asyncio import json -from typing import Dict, Generator, List, Optional, TypedDict +from typing import Dict, Generator, List, Optional import pytest -import typing_extensions from langchain_core.messages import ( AIMessage, AIMessageChunk, @@ -555,7 +554,8 @@ def test_output_matches_prompt_keys() -> None: pytest.fail(f"Response is not valid JSON: {e}") list_key = prompt_key_names["list_key"] - assert list_key in response_data, f"Expected key '{list_key}' is missing in the response." + assert list_key in response_data, \ + f"Expected key '{list_key}' is missing in the response." grocery_items = response_data[list_key] assert isinstance(grocery_items, list), f"'{list_key}' should be a list." @@ -600,4 +600,6 @@ def test_validate_response_mime_type_and_schema() -> None: response_mime_type="application/json", ).validate_environment() except ValueError as e: - pytest.fail(f"Validation failed unexpectedly with a valid MIME type and no schema: {e}") \ No newline at end of file + pytest.fail( + f"Validation failed unexpectedly with a valid MIME type and no schema: {e}" + ) \ No newline at end of file