From f21f394308f5615479fc9fa383e1918dd9a41811 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Wed, 28 Aug 2024 06:57:17 -0400 Subject: [PATCH] fix lint --- .../tests/integration_tests/test_completions_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/ai-endpoints/tests/integration_tests/test_completions_models.py b/libs/ai-endpoints/tests/integration_tests/test_completions_models.py index 3fb4e724..74a7927e 100644 --- a/libs/ai-endpoints/tests/integration_tests/test_completions_models.py +++ b/libs/ai-endpoints/tests/integration_tests/test_completions_models.py @@ -100,7 +100,9 @@ def test_basic(completions_model: str, mode: dict, func: Callable, count: int) - @pytest.mark.parametrize( "func, count", [(ainvoke, 0), (astream, 1)], ids=["ainvoke", "astream"] ) -async def test_abasic(completions_model: str, mode: dict, func: Callable, count: int) -> None: +async def test_abasic( + completions_model: str, mode: dict, func: Callable, count: int +) -> None: llm = NVIDIA(model=completions_model, **mode) response, cnt = await func(llm, "Hello, my name is") assert isinstance(response, str)