mirror of
https://github.com/vale981/ray
synced 2025-03-09 21:06:39 -04:00
e76efffec6
Following #24959, this PR moves the RL examples (online/offline/serving) into the Ray ML docs. It also splits the online and offline parts.
30 lines
No EOL
856 B
Text
30 lines
No EOL
856 B
Text
load("//bazel:python.bzl", "py_test_run_all_notebooks")
|
|
|
|
filegroup(
|
|
name = "air_examples",
|
|
srcs = glob(["*.ipynb"]),
|
|
visibility = ["//doc:__subpackages__"]
|
|
)
|
|
|
|
# --------------------------------------------------------------------
|
|
# Test all doc/source/ray-air/examples notebooks.
|
|
# --------------------------------------------------------------------
|
|
|
|
py_test_run_all_notebooks(
|
|
size = "large",
|
|
include = ["*.ipynb"],
|
|
exclude = ["huggingface_text_classification.ipynb"],
|
|
data = ["//doc/source/ray-air/examples:air_examples"],
|
|
tags = ["exclusive", "team:ml", "ray_air"],
|
|
)
|
|
|
|
|
|
# GPU Tests
|
|
|
|
py_test_run_all_notebooks(
|
|
size = "large",
|
|
include = ["huggingface_text_classification.ipynb"],
|
|
exclude = [],
|
|
data = ["//doc/source/ray-air/examples:air_examples"],
|
|
tags = ["exclusive", "team:ml", "gpu", "ray_air"],
|
|
) |