12345678910111213141516171819202122232425 |
- load("//bazel:python.bzl", "py_test_run_all_notebooks")
- filegroup(
- name="lightning_examples",
- srcs=glob(["*.ipynb"]),
- visibility=["//doc:__subpackages__"],
- )
- # GPU tests
- py_test_run_all_notebooks(
- size="large",
- include=["*.ipynb"],
- exclude=["lightning_exp_tracking.ipynb"],
- data=["//doc/source/train/examples/lightning:lightning_examples"],
- tags=["exclusive", "team:ml", "gpu", "ray_air"],
- )
- # CPU tests
- py_test_run_all_notebooks(
- size="large",
- include=["lightning_exp_tracking.ipynb"],
- exclude=[],
- data=["//doc/source/train/examples/lightning:lightning_examples"],
- tags=["exclusive", "team:ml", "ray_air"],
- )
|