Skip to content

Commit

Permalink
Removed predict return types
Browse files Browse the repository at this point in the history
The specified return types were overly resetrictive (e.g. did not include sequence labelling models)
  • Loading branch information
plonerma committed Jan 3, 2025
1 parent ddb7653 commit 8398be2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flair/nn/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -817,7 +817,7 @@ def predict(
label_name: Optional[str] = None,
return_loss: bool = False,
embedding_storage_mode: EmbeddingStorageMode = "none",
) -> Optional[Union[list[DT], tuple[float, int]]]:
):
"""Predicts the class labels for the given sentences. The labels are directly added to the sentences.
Args:
Expand Down

0 comments on commit 8398be2

Please sign in to comment.