import numpy.typing if and only if typing.TYPE_CHECKING

openfisca · Apr 23, 2021 · 5a53201 · 5a53201
1 parent 6c05349
commit 5a53201
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/STYLEGUIDE.md b/STYLEGUIDE.md
@@ -45,7 +45,8 @@ import copy
 from typing import List
 
 import numpy
-from numpy.typing import ArrayLike
+if typing.TYPE_CHECKING:
+    from numpy.typing import ArrayLike
 
 from openfisca_country_template import entities
 
@@ -75,7 +76,9 @@ from openfisca_core import axes
 from numpy import ndarray
 from copy import deepcopy
 import typing
-import numpy.typing
+if typing.TYPE_CHECKING:
+    import numpy.typing
+
 
 def do(this: typing.List) -> numpy.typing.ArrayLike:
     that = deepcopy(this)

diff --git a/openfisca_core/tracers/computation_log.py b/openfisca_core/tracers/computation_log.py
@@ -3,11 +3,11 @@
 import typing
 
 import numpy
-import numpy.typing
 
 from openfisca_core.indexed_enums import EnumArray
 
 if typing.TYPE_CHECKING:
+    import numpy.typing
     from openfisca_core.tracers import FullTracer, TraceNode
 
     Array = typing.Union[EnumArray, numpy.typing.ArrayLike]

diff --git a/openfisca_core/tracers/flat_trace.py b/openfisca_core/tracers/flat_trace.py
@@ -3,11 +3,11 @@
 import typing
 
 import numpy
-import numpy.typing
 
 from openfisca_core.indexed_enums import EnumArray
 
 if typing.TYPE_CHECKING:
+    import numpy.typing
     from openfisca_core.tracers import TraceNode, FullTracer
 
     Array = typing.Union[EnumArray, numpy.typing.ArrayLike]