Skip to content

Commit

Permalink
Resolve a bug about incompatible versions of sklearn and xgboost.
Browse files Browse the repository at this point in the history
dmlc/xgboost#11021 provides more info.
Since I don't want to use a from-source version of xgboost, Ill have to downgrade sklearn.
  • Loading branch information
elhult committed Jan 9, 2025
1 parent b55937c commit 9ff2ad1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyvista.def
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ From: nvcr.io/nvidia/jax:24.10-py3

export PIP_TMP=/tmp/pip-cache
mkdir -p $PIP_TMP
pip install --cache-dir $PIP_TMP xgboost pyvista pymc numpyro pandas numpy scipy polars openpyxl pyarrow imageio seaborn tqdm relplot xlsxwriter upsetplot fastexcel
pip install --cache-dir $PIP_TMP xgboost pyvista pymc numpyro pandas numpy scipy polars openpyxl pyarrow imageio seaborn tqdm relplot xlsxwriter upsetplot fastexcel "scikit-learn<1.6"

%test
# pytensor and matplotlib need work folders
Expand Down

0 comments on commit 9ff2ad1

Please sign in to comment.