From bdfa13beb2fa91ff0d491239792b7bd83d90d392 Mon Sep 17 00:00:00 2001 From: Simon Hawkins Date: Tue, 5 May 2020 11:08:31 +0100 Subject: [PATCH] CI/DEP: Use numba.extending.is_jitted for numba version > 0.49.0 --- pandas/core/window/numba_.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pandas/core/window/numba_.py b/pandas/core/window/numba_.py index 127957943d2ff..d6f28c9005b9e 100644 --- a/pandas/core/window/numba_.py +++ b/pandas/core/window/numba_.py @@ -1,3 +1,4 @@ +from distutils.version import LooseVersion import types from typing import Any, Callable, Dict, Optional, Tuple @@ -42,7 +43,12 @@ def make_rolling_apply( else: loop_range = range - if isinstance(func, numba.targets.registry.CPUDispatcher): + if LooseVersion(numba.__version__) >= LooseVersion("0.49.0"): + is_jitted = numba.extending.is_jitted(func) + else: + is_jitted = isinstance(func, numba.targets.registry.CPUDispatcher) + + if is_jitted: # Don't jit a user passed jitted function numba_func = func else: