From bfc5ef333a13788b7f951d4b44b8cb0816770b05 Mon Sep 17 00:00:00 2001 From: amad-person Date: Sun, 8 Nov 2020 22:02:26 +0800 Subject: [PATCH] Fix type annotation --- .../privacy/membership_inference_attack/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_privacy/privacy/membership_inference_attack/models.py b/tensorflow_privacy/privacy/membership_inference_attack/models.py index 00f9eb6..b778b7f 100644 --- a/tensorflow_privacy/privacy/membership_inference_attack/models.py +++ b/tensorflow_privacy/privacy/membership_inference_attack/models.py @@ -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: