Skip to content

Commit 58e8288

Browse files
committed
update setup.py and fix issue #1
1 parent fa85315 commit 58e8288

1 file changed

Lines changed: 10 additions & 4 deletions

File tree

setup.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -147,14 +147,20 @@ 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'],
151-
install_requires=['numpy', 'scipy', 'scikit-learn', 'pandas', 'tqdm>4.25.0', # numpy versions are for numba
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
152152
'packaging>20.9', 'slicer==0.0.7', 'numba', 'cloudpickle', 'psutil', 'shap'],
153153
extras_require=extras_require,
154154
ext_modules=ext_modules,
155155
classifiers=[
156-
'Programming Language :: Python :: 3',
157-
'Programming Language :: Python :: 3.6',
156+
"Operating System :: Microsoft :: Windows",
157+
"Operating System :: POSIX",
158+
"Operating System :: Unix",
159+
"Operating System :: MacOS",
160+
"Programming Language :: Python :: 3.7",
161+
"Programming Language :: Python :: 3.8",
162+
"Programming Language :: Python :: 3.9",
163+
"Programming Language :: Python :: 3.10",
158164
],
159165
zip_safe=False
160166
# python_requires='>3.0' we will add this at some point

0 commit comments

Comments
 (0)