Skip to content

Commit

Permalink
fix: add missing return types
Browse files Browse the repository at this point in the history
  • Loading branch information
Szymon Palucha authored and RichJackson committed Sep 18, 2024
1 parent eb9c994 commit 157c776
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions kazu/steps/ner/llm_ner.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,9 @@ def parse_result(result: str) -> Optional[dict[str, str]]:
class AzureOpenAILLMModel(LLMModel):
"""A class to interact with the Azure OpenAI API for LLMs."""

def __init__(self, model: str, deployment: str, api_version: str, sys_prompt: str, temp: float):
def __init__(
self, model: str, deployment: str, api_version: str, sys_prompt: str, temp: float
) -> None:
"""Initialize the AzureOpenAILLMModel.
:param model: The model to use.
Expand Down Expand Up @@ -117,7 +119,7 @@ def __init__(
generation_config: dict[str, Any],
location: str,
safety_settings: Optional[SafetySettingsType] = None,
):
) -> None:
"""Initialize the VertexLLMModel.
:param project: The project to use.
Expand Down

0 comments on commit 157c776

Please sign in to comment.