2021-11-10 18:00:16 -08:00
|
|
|
base_image: "anyscale/ray:nightly-py37"
|
2021-06-11 02:03:04 -07:00
|
|
|
env_vars: { }
|
|
|
|
debian_packages:
|
|
|
|
- curl
|
|
|
|
|
|
|
|
python:
|
|
|
|
pip_packages:
|
2021-10-05 09:17:33 -07:00
|
|
|
- pandas>=1.3.0 # otherwise, a version mismatch between local and remote will cause an exception
|
2021-12-05 17:57:08 -08:00
|
|
|
- git+https://github.com/ray-project/xgboost_ray.git#egg=xgboost_ray
|
2021-10-18 03:23:05 -07:00
|
|
|
- modin>=0.11.0 # ray.services has been removed
|
2021-07-07 22:37:41 -07:00
|
|
|
- s3fs
|
2021-08-31 17:07:47 +02:00
|
|
|
- fastapi
|
|
|
|
- uvicorn
|
|
|
|
- tblib
|
2021-06-11 02:03:04 -07:00
|
|
|
conda_packages: [ ]
|
|
|
|
|
|
|
|
post_build_cmds:
|
|
|
|
- pip uninstall -y ray || true
|
2021-07-07 22:37:41 -07:00
|
|
|
- pip install -U {{ env["RAY_WHEELS"] | default("ray") }}
|
2021-10-05 09:17:33 -07:00
|
|
|
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
|
2022-04-18 21:44:47 -07:00
|
|
|
# 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
|
|
|
|
|