Skip to content

Commit

Permalink
Pin xgboost for FLAML (#12432)
Browse files Browse the repository at this point in the history
Signed-off-by: harupy <[email protected]>
  • Loading branch information
harupy authored and BenWilson2 committed Jul 3, 2024
1 parent 0d6086d commit 3462aa0
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions .github/workflows/recipe.yml
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,8 @@ jobs:
- name: Install dependencies
run: |
source ./dev/install-common-deps.sh
pip install pyspark
# FLAML is incompatible with xgboost 2.1.0: https://github.com/microsoft/FLAML/issues/1314
pip install pyspark 'xgboost<2.1.0'
- name: Run tests
run: |
pytest tests/recipes
Expand All @@ -74,7 +75,8 @@ jobs:
pip install -r requirements/test-requirements.txt
pip install --no-dependencies tests/resources/mlflow-test-plugin
pip install .
pip install pyspark
# FLAML is incompatible with xgboost 2.1.0: https://github.com/microsoft/FLAML/issues/1314
pip install pyspark 'xgboost<2.1.0'
# TODO: Importing datasets in a pandas UDF (created by mlflow.pyfunc.spark_udf) crashes
# the Python worker. To avoid this, uninstall `datasets`. This is a temporary workaround.
pip uninstall -y datasets
Expand Down

0 comments on commit 3462aa0

Please sign in to comment.