From 157c77626d8063ec6a5f853c8bd1cb8f47be8000 Mon Sep 17 00:00:00 2001 From: Szymon Palucha Date: Wed, 18 Sep 2024 09:24:48 +0100 Subject: [PATCH] fix: add missing return types --- kazu/steps/ner/llm_ner.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/kazu/steps/ner/llm_ner.py b/kazu/steps/ner/llm_ner.py index e8cccbf3..0bbef123 100644 --- a/kazu/steps/ner/llm_ner.py +++ b/kazu/steps/ner/llm_ner.py @@ -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. @@ -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.