From 82029d49fa2630e73f4767903baeae9ac0013f85 Mon Sep 17 00:00:00 2001 From: Jun Gong Date: Tue, 23 May 2023 16:59:27 -0700 Subject: [PATCH] [AIR, Doc] Separate computer_vision.py into its own test of enormous size. (#35616) Signed-off-by: Jun Gong Signed-off-by: e428265 --- doc/BUILD | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/doc/BUILD b/doc/BUILD index ba0776e78cfb9..ff6dc8a956bcc 100644 --- a/doc/BUILD +++ b/doc/BUILD @@ -183,6 +183,8 @@ py_test_run_all_subdirectory( size = "large", include = ["source/ray-air/doc_code/*.py"], exclude = [ + # Too large. Use a custom test below. + "source/ray-air/doc_code/computer_vision.py", "source/ray-air/doc_code/hf_trainer.py", # Too large "source/ray-air/doc_code/predictors.py", ], @@ -190,6 +192,14 @@ py_test_run_all_subdirectory( tags = ["exclusive", "team:ml"], ) +py_test( + size = "enormous", + name = "computer_vision_test", + main = "source/ray-air/doc_code/computer_vision.py", + srcs = ["source/ray-air/doc_code/computer_vision.py"], + tags = ["exclusive", "team:ml"], +) + # -------------------------------------------------------------------- # Test all doc/source/train/doc_code code included in rst/md files. # --------------------------------------------------------------------