Add wheel to the packages of environment

feedzai · Jun 3, 2024 · e9e456a · e9e456a
1 parent 9f6ef90
commit e9e456a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.ci/test.sh b/.ci/test.sh
@@ -184,6 +184,7 @@ elif [[ $TASK == "cuda" ]]; then
 elif [[ $TASK == "mpi" ]]; then
     if [[ $METHOD == "pip" ]]; then
         cd $BUILD_DIRECTORY/python-package && python setup.py sdist || exit -1
+        pip install wheel setuptools || exit -1
         pip install --user $BUILD_DIRECTORY/python-package/dist/fairgbm-$LGB_VER.tar.gz -v --config-settings="--build-option=--mpi" || exit -1
         pytest $BUILD_DIRECTORY/tests/python_package_test || exit -1
         exit 0