diff --git a/pastas/modelstats.py b/pastas/modelstats.py index 0af90cb8..7cd6ca1d 100644 --- a/pastas/modelstats.py +++ b/pastas/modelstats.py @@ -29,6 +29,7 @@ from pandas import DataFrame from pastas.typing import Model, TimestampType + from .decorators import model_tmin_tmax from .stats import diagnostics, metrics diff --git a/pastas/recharge.py b/pastas/recharge.py index 4ae8b0f4..722030eb 100644 --- a/pastas/recharge.py +++ b/pastas/recharge.py @@ -43,6 +43,7 @@ from pandas import DataFrame from pastas.typing import ArrayLike + from .decorators import njit logger = getLogger(__name__) diff --git a/pastas/stressmodels.py b/pastas/stressmodels.py index da242ed9..b34b002c 100644 --- a/pastas/stressmodels.py +++ b/pastas/stressmodels.py @@ -24,6 +24,7 @@ from scipy.signal import fftconvolve from pastas.typing import ArrayLike, Model, Recharge, RFunc, TimestampType + from .decorators import njit, set_parameter from .recharge import Linear from .rfunc import Exponential, HantushWellModel, One