Skip to content

Commit c097d2f

Browse files
committed
fix: add numba as an installation flavour
1 parent 8f03443 commit c097d2f

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

setup.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@
3939
"unicode": [
4040
"tangled-up-in-unicode==0.2.0",
4141
],
42+
"numba": [
43+
"numba>=0.56.0, <1",
44+
]
4245
},
4346
package_data={
4447
"ydata_profiling": ["py.typed"],

src/ydata_profiling/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,8 @@
22
33
.. include:: ../../README.md
44
"""
5-
65
# ignore numba warnings
76
import warnings # isort:skip # noqa
8-
from numba.core.errors import NumbaDeprecationWarning # isort:skip # noqa
9-
10-
warnings.simplefilter("ignore", category=NumbaDeprecationWarning)
117

128
import importlib.util # isort:skip # noqa
139

@@ -23,6 +19,10 @@
2319
if spec is not None:
2420
import ydata_profiling.model.spark # isort:skip # noqa
2521

22+
spec_numba = importlib.util.find_spec("numba")
23+
if spec_numba is not None:
24+
from numba.core.errors import NumbaDeprecationWarning # isort:skip # noqa
25+
warnings.simplefilter("ignore", category=NumbaDeprecationWarning)
2626

2727
__all__ = [
2828
"pandas_decorator",

0 commit comments

Comments
 (0)