[AIR, Doc] Separate computer_vision.py into its own test of enormous …

…size. (ray-project#35616)

Signed-off-by: Jun Gong <jungong@anyscale.com>
Signed-off-by: e428265 <arvind.chandramouli@lmco.com>

doc/BUILD

@@ -183,13 +183,23 @@ 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",
     ],
     extra_srcs = [],
     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.
 # --------------------------------------------------------------------