mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
9b2cd29511
Fixes the Horovod notebook example as found in #26410 by installing Horovod in doc tests jobs. Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
39 lines
No EOL
1.1 KiB
Text
39 lines
No EOL
1.1 KiB
Text
load("//bazel:python.bzl", "py_test_run_all_notebooks")
|
|
|
|
filegroup(
|
|
name = "tune_examples",
|
|
srcs = glob(["*.ipynb"]),
|
|
visibility = ["//doc:__subpackages__"]
|
|
)
|
|
|
|
# --------------------------------------------------------------------
|
|
# Test all doc/source/tune/examples notebooks.
|
|
# --------------------------------------------------------------------
|
|
|
|
# pbt_ppo_example.ipynb is not tested right now due to large resource
|
|
# requirements
|
|
|
|
py_test_run_all_notebooks(
|
|
size = "medium",
|
|
include = ["*.ipynb"],
|
|
exclude = [
|
|
"pbt_ppo_example.ipynb",
|
|
"tune-xgboost.ipynb",
|
|
"nyc_taxi_basic_processing.ipynb", # REGRESSION
|
|
"ocr_example.ipynb", # REGRESSION
|
|
"hyperopt_example.ipynb", # REGRESSION
|
|
"sigopt_example.ipynb", # REGRESSION
|
|
"tune-sklearn.ipynb", # REGRESSION
|
|
],
|
|
data = ["//doc/source/tune/examples:tune_examples"],
|
|
tags = ["exclusive", "team:ml"],
|
|
)
|
|
|
|
# GPU tests
|
|
py_test_run_all_notebooks(
|
|
size = "large",
|
|
include = ["tune-xgboost.ipynb"],
|
|
exclude = [],
|
|
data = ["//doc/source/tune/examples:tune_examples"],
|
|
tags = ["exclusive", "team:ml", "gpu"],
|
|
) |