| 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]]] | |