Skip to content

Commit

Permalink
chore: typing
Browse files Browse the repository at this point in the history
  • Loading branch information
phil65 committed Oct 25, 2024
1 parent 681c5ea commit 65b7819
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions mkdocs_mknodes/plugin/pluginconfig.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

from collections.abc import Callable
import functools
from typing import Any

import jinjarope
from mkdocs.config import base, config_options as c
Expand Down Expand Up @@ -127,8 +128,8 @@ class PluginConfig(base.Config):
llm_model_name = c.Optional(c.Type(str))
"""LLM model name to use."""

def get_builder(self) -> Callable:
build_fn = classhelpers.to_callable(self.build_fn)
def get_builder(self) -> Callable[..., Any]:
build_fn: Callable[..., Any] = classhelpers.to_callable(self.build_fn)
build_kwargs = self.kwargs or {}
return functools.partial(build_fn, **build_kwargs)

Expand Down

0 comments on commit 65b7819

Please sign in to comment.