Skip to content

Commit e3b39b6

Browse files
committed
fix typehint of StaticEmbedding.__init__
1 parent dfef1d6 commit e3b39b6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sentence_transformers/models/StaticEmbedding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class StaticEmbedding(nn.Module):
1919
def __init__(
2020
self,
2121
tokenizer: Tokenizer | PreTrainedTokenizerFast,
22-
embedding_weights: np.array | torch.Tensor | None = None,
22+
embedding_weights: np.ndarray | torch.Tensor | None = None,
2323
embedding_dim: int | None = None,
2424
**kwargs,
2525
) -> None:
@@ -30,7 +30,7 @@ def __init__(
3030
Args:
3131
tokenizer (Tokenizer | PreTrainedTokenizerFast): The tokenizer to be used. Must be a fast tokenizer
3232
from ``transformers`` or ``tokenizers``.
33-
embedding_weights (np.array | torch.Tensor | None, optional): Pre-trained embedding weights.
33+
embedding_weights (np.ndarray | torch.Tensor | None, optional): Pre-trained embedding weights.
3434
Defaults to None.
3535
embedding_dim (int | None, optional): Dimension of the embeddings. Required if embedding_weights
3636
is not provided. Defaults to None.

0 commit comments

Comments
 (0)