Skip to content

Commit

Permalink
[Data] Test Data notebook examples (ray-project#35618)
Browse files Browse the repository at this point in the history
https://github.com/ray-project/ray/pull/33067/files#diff-23d840fadd8386e760a9e6e727860459768c58f4205391c32b6f8abd7d38a25f accidentally stopped testing Data notebook examples in CI. This PR reintroduces the tests.

Signed-off-by: Balaji Veeramani <balaji@anyscale.com>
Signed-off-by: e428265 <arvind.chandramouli@lmco.com>
  • Loading branch information
bveeramani authored and arvind-chandra committed Aug 31, 2023
1 parent 0e8d142 commit 19736c6
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions doc/source/data/examples/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,16 @@ filegroup(
visibility = ["//doc:__subpackages__"]
)

# --------------------------------------------------------------------
# Test all doc/source/data/examples notebooks.
# --------------------------------------------------------------------

py_test_run_all_notebooks(
size = "medium",
include = ["batch_inference_object_detection.ipynb", "pytorch_resnet_batch_prediction.ipynb",
"huggingface_vit_batch_prediction.ipynb"],
exclude = [],
size = "large",
include = ["*.ipynb"],
# FIXME: https://github.com/ray-project/ray/issues/36618 and
# https://github.com/ray-project/ray/issues/36619
exclude = ["ocr_example.ipynb", "nyc_taxi_basic_processing.ipynb"],
data = ["//doc/source/data/examples:data_examples"],
tags = ["exclusive", "team:data", "gpu"],
)

0 comments on commit 19736c6

Please sign in to comment.