return type fix

This commit is contained in:
Dinesh Yeduguru 2024-12-13 12:10:39 -08:00
parent 40d70864e7
commit ee5066806e

View file

@ -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 {}