We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ed15d8e commit e0d4622Copy full SHA for e0d4622
pandas/core/util/numba_.py
@@ -1,4 +1,5 @@
1
"""Common utilities for Numba operations"""
2
+from distutils.version import LooseVersion
3
import inspect
4
import types
5
from typing import Callable, Dict, Optional, Tuple
@@ -90,7 +91,12 @@ def jit_user_function(
90
91
"""
92
numba = import_optional_dependency("numba")
93
- if isinstance(func, numba.targets.registry.CPUDispatcher):
94
+ if LooseVersion(numba.__version__) >= LooseVersion("0.49.0"):
95
+ is_jitted = numba.extending.is_jitted(func)
96
+ else:
97
+ is_jitted = isinstance(func, numba.targets.registry.CPUDispatcher)
98
+
99
+ if is_jitted:
100
# Don't jit a user passed jitted function
101
numba_func = func
102
else:
0 commit comments