Fix type annotation

This commit is contained in:
amad-person 2020-11-08 22:02:26 +08:00
parent afe3944b1d
commit bfc5ef333a

View file

@ -183,7 +183,7 @@ def _get_average_ranks(logits: Iterator[np.ndarray],
def _get_ranks_for_sequence(logits: np.ndarray, def _get_ranks_for_sequence(logits: np.ndarray,
labels: np.ndarray) -> List: labels: np.ndarray) -> List[float]:
"""Returns ranks for a sequence. """Returns ranks for a sequence.
Args: Args: