added mypy config to setup.cfg

Makefile

@@ -21,7 +21,7 @@ flake8:
 
 .PHONY: mypy
 mypy:
-	mypy -p eagerpy --python-version 3.6 --ignore-missing-imports --warn-unused-ignores --warn-unused-configs --warn-return-any --warn-redundant-casts
+	mypy -p eagerpy
 
 .PHONY: install
 install:

eagerpy/astensor.py

@@ -33,11 +33,11 @@ def astensor(x: "torch.Tensor") -> PyTorchTensor:
 
 
 @overload
-def astensor(x: NativeTensor) -> Tensor:
+def astensor(x: NativeTensor) -> Tensor:  # type: ignore
     ...
 
 
-def astensor(x: Union[NativeTensor, Tensor]) -> Tensor:
+def astensor(x: Union[NativeTensor, Tensor]) -> Tensor:  # type: ignore
     if isinstance(x, Tensor):
         return x
     # we use the module name instead of isinstance
@@ -55,7 +55,7 @@ def astensor(x: Union[NativeTensor, Tensor]) -> Tensor:
     raise ValueError(f"Unknown type: {type(x)}")
 
 
-def astensors(*xs: Union[NativeTensor, Tensor]) -> Tuple[Tensor, ...]:
+def astensors(*xs: Union[NativeTensor, Tensor]) -> Tuple[Tensor, ...]:  # type: ignore
     return tuple(astensor(x) for x in xs)
 
 

eagerpy/tensor/base.py

@@ -17,7 +17,7 @@ def unwrap_(*args) -> Any:
 
 
 class BaseTensor(Tensor):
-    def __init__(self: TensorType, raw):
+    def __init__(self: TensorType, raw) -> None:
         self._raw = raw
 
     @property

eagerpy/tensor/jax.py

@@ -48,7 +48,7 @@ def unflatten(aux_data, children):
             cls._registered = True
         return super().__new__(cls)
 
-    def __init__(self, raw: "np.ndarray"):
+    def __init__(self, raw: "np.ndarray"):  # type: ignore
         global jax
         global np
         if jax is None:
@@ -57,7 +57,7 @@ def __init__(self, raw: "np.ndarray"):
         super().__init__(raw)
 
     @property
-    def raw(self) -> "np.ndarray":
+    def raw(self) -> "np.ndarray":  # type: ignore
         return super().raw
 
     @classmethod

eagerpy/tensor/numpy.py

@@ -16,11 +16,11 @@ def assert_bool(x: TensorType) -> None:
 
 
 class NumPyTensor(BaseTensor):
-    def __init__(self, raw: "np.ndarray"):
+    def __init__(self, raw: "np.ndarray"):  # type: ignore
         super().__init__(raw)
 
     @property
-    def raw(self) -> "np.ndarray":
+    def raw(self) -> "np.ndarray":  # type: ignore
         return super().raw
 
     def numpy(self: TensorType) -> Any:

eagerpy/tensor/pytorch.py

@@ -30,7 +30,7 @@ class PyTorchTensor(BaseTensor):
     def __init__(self, raw: "torch.Tensor"):
         global torch
         if torch is None:
-            torch = import_module("torch")
+            torch = import_module("torch")  # type: ignore
         super().__init__(raw)
 
     @property

eagerpy/tensor/tensor.py

@@ -15,7 +15,7 @@ class Tensor(ABC):
     __module__ = "eagerpy"
 
     @abstractmethod
-    def __init__(self, raw):
+    def __init__(self, raw) -> None:
         ...
 
     @property

eagerpy/tensor/tensorflow.py

@@ -60,14 +60,14 @@ def assert_bool(x: TensorType) -> None:
 
 
 class TensorFlowTensor(BaseTensor):
-    def __init__(self, raw: "tf.Tensor"):
+    def __init__(self, raw: "tf.Tensor"):  # type: ignore
         global tf
         if tf is None:
             tf = import_module("tensorflow")
         super().__init__(raw)
 
     @property
-    def raw(self) -> "tf.Tensor":
+    def raw(self) -> "tf.Tensor":  # type: ignore
         return super().raw
 
     def numpy(self: TensorType) -> Any:

setup.cfg

@@ -4,6 +4,28 @@ max-line-length = 80
 max-complexity = 18
 select = B,C,E,F,W,T4,B9
 
+[mypy]
+python_version = 3.6
+warn_unused_ignores = True
+warn_unused_configs = True
+warn_return_any = True
+warn_redundant_casts = True
+warn_unreachable = True
+ignore_missing_imports = False
+disallow_any_unimported = True
+
+[mypy-numpy]
+ignore_missing_imports = True
+
+[mypy-jax]
+ignore_missing_imports = True
+
+[mypy-jax.numpy]
+ignore_missing_imports = True
+
+[mypy-tensorflow]
+ignore_missing_imports = True
+
 [tool:pytest]
 filterwarnings =
     ignore::DeprecationWarning