From 9d3ed5635dbe08e9b43cf75e5afac1b6064fc67d Mon Sep 17 00:00:00 2001 From: Matt Roeschke Date: Mon, 20 Apr 2020 14:35:47 -0700 Subject: [PATCH] Use numba.extending.is_jitted for numba version > 0.50 --- pandas/core/util/numba_.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pandas/core/util/numba_.py b/pandas/core/util/numba_.py index af24189adbc27..29e74747881ae 100644 --- a/pandas/core/util/numba_.py +++ b/pandas/core/util/numba_.py @@ -1,4 +1,5 @@ """Common utilities for Numba operations""" +from distutils.version import LooseVersion import inspect import types from typing import Callable, Dict, Optional, Tuple @@ -90,7 +91,12 @@ def jit_user_function( """ numba = import_optional_dependency("numba") - 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: