mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
3fbe135a24
* Add xgboost-dask golden notebook * [examples] add modin-xgboost Jupyter notebook * Add xgboost dast gn * update modin notebook to sphinx-gallery compatible python file * fix build file * fix test * fix test * Add modin notebook anyscale connect test * Add missing file * add dask_xgboost notebook * Add the new modin golden notebook to CI * fix lint and filter out tests with py37 * Update release/golden_notebook_tests_new/golden_notebook_tests.yaml Co-authored-by: matthewdeng <matthew.j.deng@gmail.com> * Add dask, wait for cluster client, remove pytest * Replace folder * Fix * Update dask_xgboost_app_config.yaml * Update modin_xgboost_app_config.yaml * comment on filtered out tests Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
56 lines
1.2 KiB
YAML
56 lines
1.2 KiB
YAML
- name: dask_xgboost_test
|
|
owner:
|
|
mail: "antoni@anyscale.com"
|
|
slack: "@team_ml"
|
|
cluster:
|
|
app_config: dask_xgboost_app_config.yaml
|
|
compute_template: compute_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/dask_xgboost_test.py
|
|
args:
|
|
[
|
|
"--num-actors 4",
|
|
"--cpus-per-actor 4",
|
|
"--num-actors-inference 16",
|
|
"--cpus-per-actor-inference 1",
|
|
]
|
|
|
|
- name: modin_xgboost_test
|
|
owner:
|
|
mail: "antoni@anyscale.com"
|
|
slack: "@team_ml"
|
|
cluster:
|
|
app_config: modin_xgboost_app_config.yaml
|
|
compute_template: compute_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/modin_xgboost_test.py
|
|
args:
|
|
[
|
|
"--num-actors 4",
|
|
"--cpus-per-actor 4",
|
|
"--num-actors-inference 16",
|
|
"--cpus-per-actor-inference 1",
|
|
]
|
|
|
|
- name: torch_tune_serve_test
|
|
owner:
|
|
mail: "matt@anyscale.com"
|
|
slack: "@team_ml"
|
|
|
|
cluster:
|
|
app_config: torch_tune_serve_app_config.yaml
|
|
compute_template: gpu_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/torch_tune_serve_test.py
|