fixed test model

bethgelab · Oct 12, 2018 · 25b7f8e · 25b7f8e
1 parent 704f4b7
commit 25b7f8e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/foolbox/tests/test_models.py b/foolbox/tests/test_models.py
@@ -44,6 +44,9 @@ def num_classes(self):
         def predictions_and_gradient(self, image, label):
             return 'predictions', 'gradient'
 
+        def backward(self, gradient, image):
+            return image
+
     model = TestModel(bounds=(0, 1), channel_axis=1)
 
     image = np.ones((28, 28, 1), dtype=np.float32)