Skip to content

Commit 41a33b6

Browse files
committed
remove numpy restriction
1 parent 57c1a05 commit 41a33b6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

fasttreeshap/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import warnings
44
import sys
55

6-
__version__ = '0.1.2'
6+
__version__ = '0.1.3'
77

88
# check python version
99
if (sys.version_info < (3, 0)):

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ def run_setup(with_binary, with_openmp, test_xgboost, test_lightgbm, test_catboo
147147
],
148148
package_data={'fasttreeshap': ['plots/resources/*', 'cext/tree_shap.h']},
149149
cmdclass={'build_ext': build_ext},
150-
setup_requires=['numpy<1.22'], # <1.22 is because of numba
151-
install_requires=['numpy<1.22', 'scipy', 'scikit-learn', 'pandas', 'tqdm>4.25.0', # numpy versions are for numba
150+
setup_requires=['numpy'],
151+
install_requires=['numpy', 'scipy', 'scikit-learn', 'pandas', 'tqdm>4.25.0',
152152
'packaging>20.9', 'slicer==0.0.7', 'numba', 'cloudpickle', 'psutil', 'shap'],
153153
extras_require=extras_require,
154154
ext_modules=ext_modules,

0 commit comments

Comments
 (0)