Update test

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

art/attacks/evasion/adversarial_patch/adversarial_patch_tensorflow.py

@@ -177,10 +177,6 @@ def _probabilities(self, images: "tf.Tensor") -> "tf.Tensor":
             patched_input, clip_value_min=self.estimator.clip_values[0], clip_value_max=self.estimator.clip_values[1],
         )
 
-        # from matplotlib import pyplot as plt
-        # plt.matshow(patched_input[0, :, :, 0])
-        # plt.show()
-
         probabilities = self.estimator._predict_framework(patched_input)
 
         return probabilities
@@ -350,7 +346,7 @@ def generate(self, x: np.ndarray, y: Optional[np.ndarray] = None, **kwargs) -> T
 
         for _ in trange(self.max_iter, desc="Adversarial Patch TensorFlow v2"):
             for images, target in ds:
-                loss = self._train_step(images=images, target=target)
+                _ = self._train_step(images=images, target=target)
 
         return (
             self._patch.numpy(),

tests/attacks/test_adversarial_patch.py

@@ -100,16 +100,16 @@ def test_tensorflow_v2_framework(self):
             scale_max=0.41,
             learning_rate=5.0,
             batch_size=10,
-            max_iter=5,
+            max_iter=10,
             patch_shape=(28, 28, 1),
         )
 
         target = np.zeros(self.x_train_mnist.shape[0])
         patch_adv, _ = attack_ap.generate(self.x_train_mnist, target, shuffle=False)
 
-        self.assertAlmostEqual(patch_adv[8, 8, 0], 0.123679265, delta=0.05)
-        self.assertAlmostEqual(patch_adv[14, 14, 0], 0.0, delta=0.05)
-        self.assertAlmostEqual(float(np.sum(patch_adv)), 411.5689392089844, delta=1.0)
+        self.assertAlmostEqual(patch_adv[8, 8, 0], 0.21282613, delta=0.05)
+        self.assertAlmostEqual(patch_adv[14, 14, 0], 0.5411238, delta=0.05)
+        self.assertAlmostEqual(float(np.sum(patch_adv)), 378.3399658203125, delta=1.0)
 
     @unittest.skipIf(
         int(keras.__version__.split(".")[0]) == 2 and int(keras.__version__.split(".")[1]) < 3,