diff --git a/doc/source/train/examples/lightgbm/BUILD b/doc/source/train/examples/lightgbm/BUILD
index 0e4a687c7e7ff..e13b7dec19f39 100644
--- a/doc/source/train/examples/lightgbm/BUILD
+++ b/doc/source/train/examples/lightgbm/BUILD
@@ -1,15 +1,19 @@
 load("//bazel:python.bzl", "py_test_run_all_notebooks")
 
 filegroup(
-    name="lightgbm_examples",
-    srcs=glob(["*.ipynb"]),
-    visibility=["//doc:__subpackages__"],
+    name = "lightgbm_examples",
+    srcs = glob(["*.ipynb"]),
+    visibility = ["//doc:__subpackages__"],
 )
 
 py_test_run_all_notebooks(
     size = "medium",
     include = ["*.ipynb"],
+    data = ["//doc/source/train/examples/lightgbm:lightgbm_examples"],
     exclude = [],
-    data=["//doc/source/train/examples/lightgbm:lightgbm_examples"],
-    tags=["exclusive", "team:ml", "ray_air"],
-)
\ No newline at end of file
+    tags = [
+        "exclusive",
+        "ray_air",
+        "team:ml",
+    ],
+)
diff --git a/doc/source/train/examples/lightning/BUILD b/doc/source/train/examples/lightning/BUILD
index 70baa97da07cd..fb8e5fdb74dd8 100644
--- a/doc/source/train/examples/lightning/BUILD
+++ b/doc/source/train/examples/lightning/BUILD
@@ -1,19 +1,24 @@
 load("//bazel:python.bzl", "py_test_run_all_notebooks")
 
 filegroup(
-    name="lightning_examples",
-    srcs=glob(["*.ipynb"]),
-    visibility=["//doc:__subpackages__"],
+    name = "lightning_examples",
+    srcs = glob(["*.ipynb"]),
+    visibility = ["//doc:__subpackages__"],
 )
 
 # GPU tests
 py_test_run_all_notebooks(
-    size="large",
-    include=["*.ipynb"],
-    exclude=[
-        "dolly_lightning_fsdp_finetuning.ipynb", # Release Test
-        "vicuna_13b_lightning_deepspeed_finetune.ipynb", # Release Test
+    size = "large",
+    include = ["*.ipynb"],
+    data = ["//doc/source/train/examples/lightning:lightning_examples"],
+    exclude = [
+        "dolly_lightning_fsdp_finetuning.ipynb",  # Release Test
+        "vicuna_13b_lightning_deepspeed_finetune.ipynb",  # Release Test
+    ],
+    tags = [
+        "exclusive",
+        "gpu",
+        "ray_air",
+        "team:ml",
     ],
-    data=["//doc/source/train/examples/lightning:lightning_examples"],
-    tags=["exclusive", "team:ml", "gpu", "ray_air"],
 )
diff --git a/doc/source/train/examples/pytorch/BUILD b/doc/source/train/examples/pytorch/BUILD
index 8cbeabfb2e81b..5c90a4ddbc63b 100644
--- a/doc/source/train/examples/pytorch/BUILD
+++ b/doc/source/train/examples/pytorch/BUILD
@@ -3,23 +3,32 @@ load("//bazel:python.bzl", "py_test_run_all_notebooks")
 filegroup(
     name = "train_pytorch_examples",
     srcs = glob(["*.ipynb"]),
-    visibility = ["//doc:__subpackages__"]
+    visibility = ["//doc:__subpackages__"],
 )
 
 # GPU Tests
 py_test_run_all_notebooks(
     size = "large",
     include = ["*.ipynb"],
-    exclude = ["convert_existing_pytorch_code_to_ray_train.ipynb"], # CPU test
-    data=["//doc/source/train/examples/pytorch:train_pytorch_examples"],
-    tags=["exclusive", "team:ml", "gpu", "ray_air"],
+    data = ["//doc/source/train/examples/pytorch:train_pytorch_examples"],
+    exclude = ["convert_existing_pytorch_code_to_ray_train.ipynb"],  # CPU test
+    tags = [
+        "exclusive",
+        "gpu",
+        "ray_air",
+        "team:ml",
+    ],
 )
 
 # CPU Tests
 py_test_run_all_notebooks(
     size = "large",
     include = ["convert_existing_pytorch_code_to_ray_train.ipynb"],
+    data = ["//doc/source/train/examples/pytorch:train_pytorch_examples"],
     exclude = [],
-    data=["//doc/source/train/examples/pytorch:train_pytorch_examples"],
-    tags=["exclusive", "team:ml", "ray_air"],
+    tags = [
+        "exclusive",
+        "ray_air",
+        "team:ml",
+    ],
 )
diff --git a/doc/source/train/examples/transformers/BUILD b/doc/source/train/examples/transformers/BUILD
index de11f8c04d341..8175e677c1566 100644
--- a/doc/source/train/examples/transformers/BUILD
+++ b/doc/source/train/examples/transformers/BUILD
@@ -1,16 +1,21 @@
 load("//bazel:python.bzl", "py_test_run_all_notebooks")
 
 filegroup(
-    name="huggingface_transformers_examples",
-    srcs=glob(["*.ipynb"]),
-    visibility=["//doc:__subpackages__"],
+    name = "huggingface_transformers_examples",
+    srcs = glob(["*.ipynb"]),
+    visibility = ["//doc:__subpackages__"],
 )
 
 # GPU Tests
 py_test_run_all_notebooks(
     size = "large",
     include = ["*.ipynb"],
+    data = ["//doc/source/train/examples/transformers:huggingface_transformers_examples"],
     exclude = [],
-    data=["//doc/source/train/examples/transformers:huggingface_transformers_examples"],
-    tags=["exclusive", "team:ml", "gpu", "ray_air"],
-)
\ No newline at end of file
+    tags = [
+        "exclusive",
+        "gpu",
+        "ray_air",
+        "team:ml",
+    ],
+)
diff --git a/doc/source/train/examples/xgboost/BUILD b/doc/source/train/examples/xgboost/BUILD
index b9377298b3e38..fb53993770749 100644
--- a/doc/source/train/examples/xgboost/BUILD
+++ b/doc/source/train/examples/xgboost/BUILD
@@ -1,15 +1,19 @@
 load("//bazel:python.bzl", "py_test_run_all_notebooks")
 
 filegroup(
-    name="xgboost_examples",
-    srcs=glob(["*.ipynb"]),
-    visibility=["//doc:__subpackages__"],
+    name = "xgboost_examples",
+    srcs = glob(["*.ipynb"]),
+    visibility = ["//doc:__subpackages__"],
 )
 
 py_test_run_all_notebooks(
     size = "medium",
     include = ["*.ipynb"],
+    data = ["//doc/source/train/examples/xgboost:xgboost_examples"],
     exclude = [],
-    data=["//doc/source/train/examples/xgboost:xgboost_examples"],
-    tags=["exclusive", "team:ml", "ray_air"],
-)
\ No newline at end of file
+    tags = [
+        "exclusive",
+        "ray_air",
+        "team:ml",
+    ],
+)