mypy test ignore due to rare case of inheritance

bethgelab · Mar 2, 2020 · 0efe969 · 0efe969
1 parent b0c8c5a
commit 0efe969
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/foolbox/attacks/spatial_attack.py b/foolbox/attacks/spatial_attack.py
@@ -50,7 +50,7 @@ def __init__(
         self.random_steps = random_steps
 
     def __call__(  # type: ignore
-        self, model: Model, inputs: T, criterion: Union[Criterion, T],
+        self, model: Model, inputs: T, criterion: Any, **kwargs: Any,
     ) -> Tuple[T, T, T]:
         x, restore_type = ep.astensor_(inputs)
         del inputs

diff --git a/tests/test_spatial_attack.py b/tests/test_spatial_attack.py
@@ -36,6 +36,5 @@ def test_spatial_attacks(
     fmodel = fmodel.transform_bounds((0, 1))
     acc = fbn.accuracy(fmodel, x, y)
     assert acc > 0
-
-    advs, _, _ = attack(fmodel, x, y)
+    advs, _, _ = attack(fmodel, x, y)  # type: ignore
     assert fbn.accuracy(fmodel, advs, y) < acc