Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
mattf committed Aug 28, 2024
1 parent a42e389 commit f21f394
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit f21f394

Please sign in to comment.