Improve misc doc

openfisca · Aug 31, 2021 · 9a6ec1f · 9a6ec1f
1 parent 6aa9198
commit 9a6ec1f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/openfisca_core/commons/rates.py b/openfisca_core/commons/rates.py
@@ -2,10 +2,10 @@
 
 import numpy
 
-from . import ArrayLike
+from .typing import ArrayLike, NDArray
 
 
-def average_rate(target: numpy.ndarray, varying: ArrayLike[float], trim: Optional[ArrayLike[float]] = None) -> numpy.ndarray:
+def average_rate(target: NDArray[float], varying: ArrayLike[float], trim: Optional[ArrayLike[float]] = None) -> NDArray[float]:
     """Computes the average rate of a target net income.
 
     Given a ``target`` net income, and according to the ``varying`` gross
@@ -40,7 +40,7 @@ def average_rate(target: numpy.ndarray, varying: ArrayLike[float], trim: Optiona
     return average_rate
 
 
-def marginal_rate(target: numpy.ndarray, varying: numpy.ndarray, trim: Optional[ArrayLike[float]] = None) -> numpy.ndarray:
+def marginal_rate(target: NDArray[float], varying: NDArray[float], trim: Optional[ArrayLike[float]] = None) -> NDArray[float]:
     """Computes the marginal rate of a target net income.
 
     Given a ``target`` net income, and according to the ``varying`` gross