[train][docs] add highlighting to data processing quickstarts (ray-pr…

…oject#42020) Signed-off-by: Matthew Deng <[email protected]>
haiyuni · Dec 20, 2023 · 1c40c05 · 1c40c05
1 parent 915e46c
commit 1c40c05
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/doc/source/train/user-guides/data-loading-preprocessing.rst b/doc/source/train/user-guides/data-loading-preprocessing.rst
@@ -33,7 +33,8 @@ Data ingestion can be set up with four basic steps:
 
     .. tab-item:: PyTorch
 
-        .. testcode::
+        .. code-block:: python
+            :emphasize-lines: 29-32
 
             import torch
             import ray
@@ -87,11 +88,6 @@ Data ingestion can be set up with four basic steps:
             )
             result = trainer.fit()
 
-        .. testoutput::
-            :hide:
-
-            ...
-
     .. tab-item:: PyTorch Lightning
 
         .. code-block:: python
@@ -135,7 +131,7 @@ Data ingestion can be set up with four basic steps:
     .. tab-item:: HuggingFace Transformers
 
         .. code-block:: python
-            :emphasize-lines: 12,13,16,17,24,25
+            :emphasize-lines: 12,13,16,17,23,29,30,34
 
             import ray
             import ray.train