[Release] Fix XGBoost Golden Notebook Tests (#23996)

Xgboost released a new version a few days ago. Due to caching of the Anyscale cluster env, this resulted in the server having an outdated xgboost version while the client has the most recent version causing the test to fail.

Instead, we reinstall xgboost-ray and xgboost in the post build commands so that these dependencies are not being cached in the cluster env.
This commit is contained in:
Amog Kamsetty 2022-04-18 21:44:47 -07:00 committed by GitHub
parent 1c3329fa38
commit 9ec5793bea
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 0 deletions

View file

@ -17,3 +17,8 @@ post_build_cmds:
- pip uninstall -y ray || true
- pip install -U {{ env["RAY_WHEELS"] | default("ray") }}
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
# Upgrade the XGBoost-Ray version in post build commands, otherwise it will be cached in the Anyscale Docker image.
- echo {{ env["TIMESTAMP"] }}
- pip install -U xgboost # Upgrade to latest xgboost version so cached version is not used.
- pip uninstall xgboost_ray -y # Uninstall first so pip does a reinstall.
- pip install -U --no-cache-dir git+https://github.com/ray-project/xgboost_ray.git#egg=xgboost_ray

View file

@ -18,3 +18,9 @@ post_build_cmds:
- pip uninstall -y ray || true
- pip install -U {{ env["RAY_WHEELS"] | default("ray") }}
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
# Upgrade the XGBoost-Ray version in post build commands, otherwise it will be cached in the Anyscale Docker image.
- echo {{ env["TIMESTAMP"] }}
- pip install -U xgboost # Upgrade to latest xgboost version so cached version is not used.
- pip uninstall xgboost_ray -y # Uninstall first so pip does a reinstall.
- pip install -U --no-cache-dir git+https://github.com/ray-project/xgboost_ray.git#egg=xgboost_ray