mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -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>
114 lines
3.3 KiB
Text
114 lines
3.3 KiB
Text
# --------------------------------------------------------------------
|
|
# Tests from the doc/examples directory.
|
|
# Please keep these sorted alphabetically, but start with the
|
|
# root directory.
|
|
# --------------------------------------------------------------------
|
|
|
|
# Support for Dask has been dropped in 3.6.
|
|
py_test(
|
|
name = "dask_xgboost",
|
|
size = "medium",
|
|
main = "examples/dask_xgboost/dask_xgboost.py",
|
|
srcs = ["examples/dask_xgboost/dask_xgboost.py"],
|
|
tags = ["exclusive", "team:ml", "py37"],
|
|
args = ["--smoke-test", "--address ''", "--num-actors 4",
|
|
"--cpus-per-actor 1", "--num-actors-inference 4",
|
|
"--cpus-per-actor-inference 1"]
|
|
)
|
|
|
|
# Support for Modin has been dropped in 3.6.
|
|
py_test(
|
|
name = "modin_xgboost",
|
|
size = "medium",
|
|
main = "examples/modin_xgboost/modin_xgboost.py",
|
|
srcs = ["examples/modin_xgboost/modin_xgboost.py"],
|
|
tags = ["exclusive", "team:ml", "py37"],
|
|
args = ["--smoke-test", "--address ''", "--num-actors 4",
|
|
"--cpus-per-actor 1", "--num-actors-inference 4",
|
|
"--cpus-per-actor-inference 1"]
|
|
)
|
|
|
|
py_test(
|
|
name = "plot_hyperparameter",
|
|
size = "small",
|
|
srcs = ["examples/plot_hyperparameter.py"],
|
|
tags = ["exclusive", "team:ml"]
|
|
)
|
|
|
|
py_test(
|
|
name = "plot_parameter_server",
|
|
size = "medium",
|
|
srcs = ["examples/plot_parameter_server.py"],
|
|
tags = ["exclusive", "team:ml"]
|
|
)
|
|
|
|
py_test(
|
|
name = "plot_pong_example",
|
|
size = "large",
|
|
srcs = ["examples/plot_pong_example.py"],
|
|
tags = ["exclusive", "team:ml"]
|
|
)
|
|
|
|
py_test(
|
|
name = "progress_bar",
|
|
size = "small",
|
|
srcs = ["examples/progress_bar.py"],
|
|
tags = ["exclusive", "team:ml"]
|
|
)
|
|
|
|
|
|
# Directory: examples/doc_code
|
|
py_test(
|
|
name = "doc_code_tf_example",
|
|
size = "small",
|
|
main = "examples/doc_code/tf_example.py",
|
|
srcs = ["examples/doc_code/tf_example.py"],
|
|
tags = ["exclusive", "tf", "team:ml"]
|
|
)
|
|
|
|
py_test(
|
|
name = "doc_code_torch_example",
|
|
size = "small",
|
|
main = "examples/doc_code/torch_example.py",
|
|
srcs = ["examples/doc_code/torch_example.py"],
|
|
tags = ["exclusive", "pytorch", "team:ml"]
|
|
)
|
|
|
|
py_test(
|
|
name = "doc_code_metrics_example",
|
|
size = "small",
|
|
main = "examples/doc_code/metrics_example.py",
|
|
srcs = ["examples/doc_code/metrics_example.py"],
|
|
tags = ["exclusive", "team:serve"]
|
|
)
|
|
|
|
py_test(
|
|
name = "doc_code_runtime_env_example",
|
|
size = "small",
|
|
main = "examples/doc_code/runtime_env_example.py",
|
|
srcs = ["examples/doc_code/runtime_env_example.py"],
|
|
tags = ["exclusive", "post_wheel_build", "team:serve"]
|
|
)
|
|
|
|
# --------------------------------------------------------------------
|
|
# Tests from the doc/source/tune/_tutorials directory.
|
|
# Please keep these sorted alphabetically.
|
|
# --------------------------------------------------------------------
|
|
|
|
py_test(
|
|
name = "tune_sklearn",
|
|
size = "medium",
|
|
main = "source/tune/_tutorials/tune-sklearn.py",
|
|
srcs = ["source/tune/_tutorials/tune-sklearn.py"],
|
|
tags = ["exclusive", "example", "team:ml"],
|
|
args = ["--smoke-test"]
|
|
)
|
|
|
|
py_test(
|
|
name = "tune_serve_integration_mnist",
|
|
size = "medium",
|
|
main = "source/tune/_tutorials/tune-serve-integration-mnist.py",
|
|
srcs = ["source/tune/_tutorials/tune-serve-integration-mnist.py"],
|
|
tags = ["exclusive", "example", "team:ml"],
|
|
args = ["--smoke-test", "--from_scratch", "--day 0"]
|
|
)
|