forked from 626_privacy/tensorflow_privacy
Fix type annotation
This commit is contained in:
parent
afe3944b1d
commit
bfc5ef333a
1 changed files with 1 additions and 1 deletions
|
@ -183,7 +183,7 @@ def _get_average_ranks(logits: Iterator[np.ndarray],
|
|||
|
||||
|
||||
def _get_ranks_for_sequence(logits: np.ndarray,
|
||||
labels: np.ndarray) -> List:
|
||||
labels: np.ndarray) -> List[float]:
|
||||
"""Returns ranks for a sequence.
|
||||
|
||||
Args:
|
||||
|
|
Loading…
Reference in a new issue