From ee5066806eddec68a9a734dbd06b74d7b4d081c7 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Fri, 13 Dec 2024 12:10:39 -0800 Subject: [PATCH] return type fix --- .../inline/inference/sentence_transformers/config.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/inference/sentence_transformers/config.py b/llama_stack/providers/inline/inference/sentence_transformers/config.py index 89f34f7e0..53f17cfd5 100644 --- a/llama_stack/providers/inline/inference/sentence_transformers/config.py +++ b/llama_stack/providers/inline/inference/sentence_transformers/config.py @@ -4,11 +4,13 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. +from typing import Any, Dict + from pydantic import BaseModel class SentenceTransformersInferenceConfig(BaseModel): @classmethod - def sample_run_config(cls) -> "SentenceTransformersInferenceConfig": + def sample_run_config(cls) -> Dict[str, Any]: return {}