Skip to content

Commit

Permalink
Change embedding model return type to Callable
Browse files Browse the repository at this point in the history
Co-authored-by: Simon <80467011+sorgfresser@users.noreply.github.com>
  • Loading branch information
juanmc2005 and sorgfresser authored Oct 30, 2023
1 parent 679dee4 commit 26a1417
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diart/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def __init__(self, model_info, hf_token: Union[Text, bool, None] = True):
self.model_info = model_info
self.hf_token = hf_token

def __call__(self) -> Union[Model, PretrainedSpeakerEmbedding]:
def __call__(self) -> Callable:
try:
return Model.from_pretrained(self.model_info, use_auth_token=self.hf_token)
except HTTPError:
Expand Down

0 comments on commit 26a1417

Please sign in to comment.