Fix mypy warning on typing

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Jan 16, 2025 · 911884b · 911884b
1 parent 0aa91b5
commit 911884b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/art/attacks/evasion/adversarial_patch/adversarial_patch_pytorch.py b/art/attacks/evasion/adversarial_patch/adversarial_patch_pytorch.py
@@ -513,7 +513,9 @@ def generate(  # type: ignore
                 logger.info("Setting labels to estimator classification predictions.")
                 y = to_categorical(np.argmax(self.estimator.predict(x=x), axis=1), nb_classes=self.estimator.nb_classes)
 
-            y = check_and_transform_label_format(labels=y, nb_classes=self.estimator.nb_classes)
+                y_array: np.ndarray = y
+
+                y = check_and_transform_label_format(labels=y_array, nb_classes=self.estimator.nb_classes)
 
             # check if logits or probabilities
             y_pred = self.estimator.predict(x=x[[0]])