typing

Signed-off-by: Taesung Lee <[email protected]>

Author: TS-Lee

art/attacks/poisoning/gradient_matching_attack.py

@@ -22,7 +22,7 @@
 from __future__ import absolute_import, division, print_function, unicode_literals
 
 import logging
-from typing import Dict, Tuple, TYPE_CHECKING, List
+from typing import Any, Dict, Tuple, TYPE_CHECKING, List
 
 import numpy as np
 from tqdm.auto import trange, tqdm
@@ -440,7 +440,7 @@ def poison(
         x_train[best_indices_poison] = best_x_poisoned
         return x_train, y_train  # y_train has not been modified.
 
-    def __poison__pytorch(self, x_poison: np.ndarray, y_poison: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
+    def __poison__pytorch(self, x_poison: np.ndarray, y_poison: np.ndarray) -> Tuple[Any, float]:
         """
         Optimize the poison by matching the gradient within the perturbation budget.