Skip to content

Commit

Permalink
mypy
Browse files Browse the repository at this point in the history
Signed-off-by: Henry Lindeman <[email protected]>
  • Loading branch information
HenryL27 committed Feb 27, 2025
1 parent d93366e commit 05c9892
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lib/sycamore/sycamore/llms/anthropic.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 05c9892

Please sign in to comment.