From 05c9892ef41ede9f5443298f50f6819bc4e159d7 Mon Sep 17 00:00:00 2001 From: Henry Lindeman Date: Thu, 27 Feb 2025 15:57:10 -0800 Subject: [PATCH] mypy Signed-off-by: Henry Lindeman --- lib/sycamore/sycamore/llms/anthropic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/sycamore/sycamore/llms/anthropic.py b/lib/sycamore/sycamore/llms/anthropic.py index 217396b26..ee48b3e50 100644 --- a/lib/sycamore/sycamore/llms/anthropic.py +++ b/lib/sycamore/sycamore/llms/anthropic.py @@ -235,7 +235,7 @@ def generate_batch(self, *, prompts: list[RenderedPrompt], llm_kwargs: Optional[ kwargs = get_generate_kwargs(p, llm_kwargs) kwargs["model"] = self.model.value kwargs["max_tokens"] = kwargs.get("max_tokens", 1024) - mparams = MessageCreateParamsNonStreaming(**kwargs) + mparams = MessageCreateParamsNonStreaming(**kwargs) # type: ignore rq = Request(custom_id=str(i), params=mparams) calls.append(rq)