Merge pull request #561 from bethgelab/numba_improvements

updated numba version and fixed numba warnings

Author: jonasrauber

foolbox/attacks/brendel_bethge.py

@@ -22,7 +22,7 @@
 
 
 try:
-    from numba import jitclass  # type: ignore
+    from numba.experimental import jitclass  # type: ignore
     import numba
 except (ModuleNotFoundError, ImportError) as e:  # pragma: no cover
     # delay the error until the attack is initialized

requirements.txt

@@ -7,4 +7,3 @@ codecov==2.0.15
 coveralls==1.10.0
 mypy==0.770
 pre-commit==2.3.0
-responses==0.10.9

setup.cfg

@@ -51,6 +51,9 @@ ignore_missing_imports = True
 [mypy-responses.*]
 ignore_missing_imports = True
 
+[mypy-numba]
+ignore_missing_imports = True
+
 [tool:pytest]
 addopts =
     --verbose

tests/requirements.txt

@@ -3,7 +3,8 @@ torchvision==0.5.0
 jaxlib==0.1.46
 jax==0.1.66
 tensorflow==2.0.1
-numba==0.47.0
+numba==0.50.1
 matplotlib==3.2.1
 pillow==7.1.2
 tensorboardX==2.0
+responses==0.10.9