Files
2026-07-13 13:17:40 +08:00

380 lines
12 KiB
YAML

group: data tests
depends_on:
- forge
- oss-ci-base_ml-multipy(*)
- ray-core-build(*)
- ray-dashboard-build
steps:
# builds
- name: data17build-multipy
label: "wanda: data17build-py{{array.python}}"
wanda: ci/docker/data17.build.wanda.yaml
array:
python:
- "3.10"
env:
PYTHON: "{{array.python}}"
tags: cibase
- name: datalbuild-multipy
label: "wanda: datalbuild-py{{array.python}}"
wanda: ci/docker/datal.build.wanda.yaml
array:
python:
- "3.10"
- "3.12"
env:
PYTHON: "{{array.python}}"
tags: cibase
- name: datanbuild-multipy
wanda: ci/docker/datan.build.wanda.yaml
env:
PYTHON: "3.10"
tags: cibase
- name: datamongobuild-multipy
label: "wanda: datamongobuild-py3.10"
wanda: ci/docker/datamongo.build.wanda.yaml
env:
PYTHON: "3.10"
tags: cibase
- name: datatfdsbuild-multipy
label: "wanda: datatfdsbuild-py3.12"
wanda: ci/docker/datatfds.build.wanda.yaml
env:
PYTHON: "3.12"
tags: cibase
- name: docgpubuild-py312
label: "wanda: docgpubuild-py3.12"
wanda: ci/docker/docgpu.build.wanda.yaml
env:
PYTHON: "3.12"
depends_on: oss-ci-base_gpu-multipy(python=3.12)
tags: cibase
# type check
- label: ":python: pyrefly type check"
tags:
- data
instance_type: small
commands:
- bash ci/lint/pyrefly-check.sh
depends_on: data17build-multipy(python=3.10)
job_env: data17build-py3.10
# tests
- label: ":database: data: arrow v17 tests"
tags:
- data
instance_type: large
parallelism: 8
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers "$${BUILDKITE_PARALLEL_JOB_COUNT}"
--worker-id "$${BUILDKITE_PARALLEL_JOB}" --parallelism-per-worker 3
--build-name data17build-py3.10 --python-version 3.10
--except-tags data_integration,doctest,data_non_parallel,dask,needs_credentials,tensorflow_datasets,cudf
depends_on: data17build-multipy(python=3.10)
- label: ":database: data: tensorflow-datasets tests"
key: data_tensorflow_datasets_tests
tags:
- data
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... data
--build-name datatfdsbuild-py3.12 --python-version 3.12
--only-tags tensorflow_datasets
depends_on: datatfdsbuild-multipy
- label: ":database: data: arrow v17 tests (data_non_parallel)"
tags:
- data
- data_non_parallel
instance_type: medium
parallelism: 3
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers "$${BUILDKITE_PARALLEL_JOB_COUNT}"
--worker-id "$${BUILDKITE_PARALLEL_JOB}"
--build-name data17build-py3.10 --python-version 3.10
--only-tags data_non_parallel
--except-tags data_non_parallel_postmerge_only,cudf
depends_on: data17build-multipy(python=3.10)
- label: ":database: data: arrow v23 tests"
tags:
- data
- skip-on-premerge
instance_type: medium
parallelism: 8
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers "$${BUILDKITE_PARALLEL_JOB_COUNT}"
--worker-id "$${BUILDKITE_PARALLEL_JOB}" --parallelism-per-worker 3
--build-name datalbuild-py3.10 --python-version 3.10
--except-tags data_integration,doctest,data_non_parallel,dask,needs_credentials,tensorflow_datasets,cudf
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: arrow v23 tests (data_non_parallel)"
tags:
- data
- data_non_parallel
- skip-on-premerge
instance_type: medium
parallelism: 3
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers "$${BUILDKITE_PARALLEL_JOB_COUNT}"
--worker-id "$${BUILDKITE_PARALLEL_JOB}"
--build-name datalbuild-py3.10 --python-version 3.10
--only-tags data_non_parallel
--except-tags data_non_parallel_postmerge_only,cudf
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: arrow v23 py{{array.python}} tests ({{array.worker_id}})"
key: datal_python_tests
if: build.pull_request.labels includes "continuous-build" || pipeline.id == "0189e759-8c96-4302-b6b5-b4274406bf89" || pipeline.id == "018f4f1e-1b73-4906-9802-92422e3badaa"
tags:
- data
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers 2 --worker-id {{array.worker_id}} --parallelism-per-worker 3
--build-name datalbuild-py{{array.python}} --python-version {{array.python}}
--except-tags data_integration,doctest,data_non_parallel,dask,needs_credentials,tensorflow_datasets,cudf
depends_on: datalbuild-multipy($)
array:
python: ["3.12"]
worker_id: ["0", "1"]
- label: ":database: data: arrow v23 py{{array.python}} tests (data_non_parallel)"
key: datal_python_non_parallel_tests
if: build.pull_request.labels includes "continuous-build" || pipeline.id == "0189e759-8c96-4302-b6b5-b4274406bf89" || pipeline.id == "018f4f1e-1b73-4906-9802-92422e3badaa"
tags:
- data
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--build-name datalbuild-py{{array.python}} --python-version {{array.python}}
--only-tags data_non_parallel
--except-tags data_non_parallel_postmerge_only,cudf
depends_on: datalbuild-multipy($)
array:
python: ["3.12"]
- label: ":database: data: arrow nightly tests"
tags:
- data
- skip-on-premerge
- oss
- disabled
instance_type: medium
parallelism: 2
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--workers "$${BUILDKITE_PARALLEL_JOB_COUNT}"
--worker-id "$${BUILDKITE_PARALLEL_JOB}" --parallelism-per-worker 3
--build-name datanbuild-py3.10 --python-version 3.10
--except-tags data_integration,doctest,data_non_parallel,dask,needs_credentials,tensorflow_datasets,cudf
depends_on: datanbuild-multipy
soft_fail: true
- label: ":database: data: arrow nightly tests (data_non_parallel)"
tags:
- data
- skip-on-premerge
- oss
- disabled
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--build-name datanbuild-py3.10 --python-version 3.10
--only-tags data_non_parallel
--except-tags data_non_parallel_postmerge_only,cudf
depends_on: datanbuild-multipy
soft_fail: true
- label: ":database: data: postmerge arrow v17 tests"
tags:
- data
- skip-on-premerge
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--build-name data17build-py3.10 --python-version 3.10
--only-tags data_non_parallel_postmerge_only
depends_on: data17build-multipy(python=3.10)
- label: ":database: data: postmerge arrow v23 tests"
tags:
- data
- skip-on-premerge
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //python/ray/air/... data
--build-name datalbuild-py3.10 --python-version 3.10
--only-tags data_non_parallel_postmerge_only
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: dask tests"
tags:
- data
- dask
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... data
--build-name datalbuild-py3.12 --python-version 3.12
--only-tags dask
depends_on: datalbuild-multipy(python=3.12)
- label: ":database: data: doc tests"
if: '!(build.pull_request.labels includes "docs-go")'
tags:
- data
- doc
instance_type: medium
commands:
# doc tests
- bazel run //ci/ray_ci:test_in_docker -- python/ray/... //doc/... data
--build-name datalbuild-py3.10
--python-version 3.10
--except-tags gpu
--only-tags doctest
--parallelism-per-worker 2
# doc examples
- bazel run //ci/ray_ci:test_in_docker -- //doc/... data
--build-name datalbuild-py3.10
--python-version 3.10
--except-tags gpu,post_wheel_build,doctest,dask
--parallelism-per-worker 2
--skip-ray-installation
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: dask doc tests"
if: '!(build.pull_request.labels includes "docs-go")'
tags:
- data
- doc
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //doc/... data
--build-name datalbuild-py3.12
--python-version 3.12
--only-tags dask
--parallelism-per-worker 2
depends_on: datalbuild-multipy(python=3.12)
- label: ":database: data: doc gpu tests"
if: '!(build.pull_request.labels includes "docs-go")'
tags:
- data
- doc
- gpu
instance_type: gpu-large
commands:
# doc tests
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... //doc/... data
--build-name docgpubuild-py3.10
--only-tags doctest
--except-tags cpu
--python-version 3.10
# doc examples
- bazel run //ci/ray_ci:test_in_docker -- //doc/... data
--build-name docgpubuild-py3.10
--except-tags doctest
--only-tags gpu
--skip-ray-installation
--python-version 3.10
depends_on: docgpubuild
- label: ":database: data: cuDF batch format tests"
tags:
- data
- gpu
- cudf
instance_type: gpu-large
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... data
--build-name docgpubuild-py3.12
--only-tags cudf
--gpus 1
--python-version 3.12
depends_on: docgpubuild-py312
- label: ":database: data: integration tests"
tags:
- data
instance_type: medium
commands:
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... data
--build-name datamongobuild-py3.10
--python-version 3.10
--build-type java
--only-tags data_integration
--except-tags doctest
depends_on: datamongobuild-multipy
- label: ":database: data: dashboard tests"
tags:
- data
- dashboard
instance_type: small
commands:
- bazel run //ci/ray_ci:test_in_docker -- python/ray/dashboard/... data
--build-name datalbuild-py3.10
--python-version 3.10
--parallelism-per-worker 3
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: flaky tests"
key: data_flaky_tests
tags:
- data
- flaky
- skip-on-premerge
instance_type: medium
soft_fail: true
commands:
- bazel run //ci/ray_ci:test_in_docker -- //... data --run-flaky-tests
--parallelism-per-worker 3
--build-name datalbuild-py3.10
--python-version 3.10
--except-tags gpu_only,gpu
depends_on: datalbuild-multipy(python=3.10)
- label: ":database: data: flaky gpu tests"
key: data_flaky_gpu_tests
tags:
- data
- skip-on-premerge
instance_type: gpu-large
soft_fail: true
commands:
- bazel run //ci/ray_ci:test_in_docker -- //... data --run-flaky-tests
--build-name docgpubuild-py3.10
--only-tags gpu,gpu_only
--python-version 3.10
depends_on: docgpubuild
- label: ":database: data: postmerge authenticated tests"
key: data_postmerge_authenticated_tests
tags:
- data
- oss
- skip-on-premerge
instance_type: medium
commands:
- $(python ci/env/setup_credentials.py)
- bazel run //ci/ray_ci:test_in_docker -- //python/ray/data/... data
--build-name datalbuild-py3.10
--python-version 3.10
--only-tags needs_credentials
--test-env=SNOWFLAKE_USER --test-env=SNOWFLAKE_ACCOUNT
--test-env=SNOWFLAKE_DATABASE --test-env=SNOWFLAKE_SCHEMA
--test-env=SNOWFLAKE_WAREHOUSE --test-env=SNOWFLAKE_PRIVATE_KEY
depends_on: datalbuild-multipy(python=3.10)