2022-05-17 17:03:12 +01:00
|
|
|
base_image: {{ env["RAY_IMAGE_NIGHTLY_CPU"] | default("anyscale/ray:nightly-py37") }}
|
2021-06-11 02:03:04 -07:00
|
|
|
env_vars: { }
|
|
|
|
debian_packages:
|
|
|
|
- curl
|
|
|
|
|
|
|
|
python:
|
|
|
|
pip_packages:
|
2021-08-31 17:07:47 +02: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-06-11 02:03:04 -07:00
|
|
|
- dask
|
2021-08-31 17:07:47 +02:00
|
|
|
- fastapi
|
|
|
|
- uvicorn
|
|
|
|
- tblib
|
2021-06-11 02:03:04 -07:00
|
|
|
conda_packages: [ ]
|
|
|
|
|
|
|
|
post_build_cmds:
|
2022-05-22 16:07:03 +09:00
|
|
|
- pip uninstall -y ray || true && pip install -U {{ env["RAY_WHEELS"] | default("ray") }}
|
2021-09-10 17:50:31 +01: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"] }}
|
2022-05-31 13:46:17 +02:00
|
|
|
- pip install -U --force-reinstall --no-deps --no-cache-dir git+https://github.com/ray-project/xgboost_ray.git#egg=xgboost_ray xgboost
|
|
|
|
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
|