use TYPE_CHECKING instead of False

eagerpy/modules.py

@@ -1,6 +1,7 @@
 from importlib import import_module
 import inspect
 from types import ModuleType
+from typing import Any
 import functools
 
 from .astensor import astensor
@@ -32,7 +33,7 @@ def __dir__(self):
         # makes sure tab completion works
         return import_module(self.__name__).__dir__()
 
-    def __getattr__(self, name):
+    def __getattr__(self, name: str) -> Any:
         attr = getattr(import_module(self.__name__), name)
         if callable(attr):
             attr = wrap(attr)

eagerpy/tensor/jax.py

@@ -5,17 +5,18 @@
 
 # from .tensor import Tensor
 
-from typing import Tuple, cast, Union, Any, TypeVar
+from typing import Tuple, cast, Union, Any, TypeVar, TYPE_CHECKING
 from importlib import import_module
 from collections.abc import Iterable
 import numpy as onp
 
 
-if False:
+if TYPE_CHECKING:
     # for static analyzers
     import jax
     import jax.numpy as np
 else:
+    # lazy import in JAXTensor
     jax = None
     np = None
 

eagerpy/tensor/pytorch.py

@@ -7,14 +7,15 @@
 
 import numpy as np
 from collections.abc import Iterable
-from typing import Tuple, cast, Union, Any, TypeVar
+from typing import Tuple, cast, Union, Any, TypeVar, TYPE_CHECKING
 from importlib import import_module
 
 
-if False:
+if TYPE_CHECKING:
     import torch  # for static analyzers
 else:
-    torch = None  # type: ignore
+    # lazy import in PyTorchTensor
+    torch = None
 
 
 Tensor = TypeVar("Tensor", bound="PyTorchTensor")

eagerpy/tensor/tensorflow.py

@@ -7,7 +7,7 @@
 from .. import index
 
 import functools
-from typing import Tuple, cast, Union, Any, TypeVar, Callable
+from typing import Tuple, cast, Union, Any, TypeVar, Callable, TYPE_CHECKING
 from importlib import import_module
 from collections.abc import Iterable
 import numpy as np
@@ -17,9 +17,10 @@
 F = TypeVar("F", bound=FuncType)
 
 
-if False:
+if TYPE_CHECKING:
     import tensorflow as tf  # for static analyzers
 else:
+    # lazy import in TensorFlowTensor
     tf = None
 
 

setup.cfg

@@ -18,6 +18,6 @@ exclude_lines =
     # Have to re-enable the standard pragma
     pragma: no cover
 
-    # Don't complain if tests don't hit defensive assertion code:
     @abstractmethod
     @overload
+    TYPE_CHECKING