Apppublic

BridgeAI-Lab/Sem-nCG

sourceHugging Faceupdated 2y agoView on Hugging Face

1likes

type_aliases.py10 linesDownload Raw Back to root

1from typing import List, Union2 3from numpy.typing import NDArray4 5NumSentencesType = Union[List[int], List[List[int]]]6EmbeddingSlicesType = Union[List[NDArray], List[List[NDArray]]]7DEVICE_TYPE = Union[bool, str, int, List[Union[str, int]]]8ENCODER_DEVICE_TYPE = Union[str, int, List[Union[str, int]]]9DOCUMENT_TYPE = Union[List[str], List[List[str]]]10