File size: 429 Bytes
a249916 |
1 2 3 4 5 6 7 8 9 10 11 |
from typing import List, Union
from numpy.typing import NDArray
NumSentencesType = Union[List[int], List[List[int]]]
EmbeddingSlicesType = Union[List[NDArray], List[List[NDArray]]]
PREDICTION_TYPE = Union[List[str], List[List[str]]]
REFERENCE_TYPE = Union[List[str], List[List[str]], List[List[List[str]]]]
DEVICE_TYPE = Union[bool, str, int, List[Union[str, int]]]
ENCODER_DEVICE_TYPE = Union[str, int, List[Union[str, int]]]
|