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,
|
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:
|
||||||
|
|
Loading…
Reference in a new issue