mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
6c5229295e
OSS release tests currently run with hardcoded Python 3.7 base. In the future we will want to run tests on different python versions. This PR adds support for a new `python` field in the test configuration. The python field will determine both the base image used in the Buildkite runner docker container (for Ray client compatibility) and the base image for the Anyscale cluster environments. Note that in Buildkite, we will still only wait for the python 3.7 base image before kicking off tests. That is acceptable, as we can assume that most wheels finish in a similar time, so even if we wait for the 3.7 image and kick off a 3.8 test, that runner will wait maybe for 5-10 more minutes.
26 lines
1 KiB
YAML
Executable file
26 lines
1 KiB
YAML
Executable file
base_image: {{ env["RAY_IMAGE_NIGHTLY_CPU"] | default("anyscale/ray:nightly-py37") }}
|
|
env_vars: { }
|
|
debian_packages:
|
|
- curl
|
|
|
|
python:
|
|
pip_packages:
|
|
- pandas>=1.3.0 # otherwise, a version mismatch between local and remote will cause an exception
|
|
- git+https://github.com/ray-project/xgboost_ray.git#egg=xgboost_ray
|
|
- modin>=0.11.0 # ray.services has been removed
|
|
- s3fs
|
|
- fastapi
|
|
- uvicorn
|
|
- tblib
|
|
conda_packages: [ ]
|
|
|
|
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
|
|
|