@@ -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