From e9009fbff2c762f969cd08838eaba6dfaa4b51fc Mon Sep 17 00:00:00 2001 From: Taher Chegini Date: Mon, 12 Feb 2024 10:24:21 -0500 Subject: [PATCH] ENH: Use njit instead of jit. [skip ci] --- hydrosignatures/hydrosignatures.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/hydrosignatures/hydrosignatures.py b/hydrosignatures/hydrosignatures.py index 50bca12..1dbc066 100644 --- a/hydrosignatures/hydrosignatures.py +++ b/hydrosignatures/hydrosignatures.py @@ -1,5 +1,5 @@ """Function for computing hydrologic signature.""" -# pyright: reportGeneralTypeIssues=false, reportAttributeAccessIssue=false +# pyright: reportGeneralTypeIssues=false from __future__ import annotations import calendar @@ -24,10 +24,10 @@ try: from numba import config as numba_config - from numba import jit, prange + from numba import njit, prange - ngjit = functools.partial(jit, nopython=True, nogil=True) # pyright: ignore[reportAssignmentType] - numba_config.THREADING_LAYER = "workqueue" # pyright: ignore[reportGeneralTypeIssues] + ngjit = functools.partial(njit, nogil=True) # pyright: ignore[reportAssignmentType] + numba_config.THREADING_LAYER = "workqueue" has_numba = True except ImportError: has_numba = False