update doc build rule to exclude hf_trainer.py properly. (ray-project…

…#37837) Signed-off-by: xwjiang2010 <[email protected]> Signed-off-by: NripeshN <[email protected]>
NripeshN · Aug 15, 2023 · db4a116 · db4a116
1 parent 98c4dad
commit db4a116
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/doc/BUILD b/doc/BUILD
@@ -203,7 +203,6 @@ py_test_run_all_subdirectory(
  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
  ],
  extra_srcs = [],
  tags = ["exclusive", "team:ml"],
@@ -224,7 +223,10 @@ py_test(
 py_test_run_all_subdirectory(
  size = "large",
  include = ["source/train/doc_code/*.py"],
- exclude = ["source/train/doc_code/predictors.py"], # too large
+ exclude = [
+ "source/train/doc_code/predictors.py", # Too large
+ "source/train/doc_code/hf_trainer.py", # Too large
+ ],
  extra_srcs = [],
  tags = ["exclusive", "team:ml"],
 )