File size: 353 Bytes
27a1559
a54024a
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from typing import List, Union

from numpy.typing import NDArray

NumSentencesType = Union[List[int], List[List[int]]]
EmbeddingSlicesType = Union[List[NDArray], List[List[NDArray]]]
DEVICE_TYPE = Union[bool, str, int, List[Union[str, int]]]
ENCODER_DEVICE_TYPE = Union[str, int, List[Union[str, int]]]
DOCUMENT_TYPE = Union[List[str], List[List[str]]]