[train][doc] Rename checkpoint_upload_func -> checkpoint_upload_fn in docs (ray-project#60390)

Signed-off-by: Timothy Seah <tseah@anyscale.com>
Signed-off-by: peterxcli <peterxcli@gmail.com>

Author: TimothySeah

doc/source/train/doc_code/checkpoints.py

@@ -503,7 +503,7 @@ def train_fn(config):
 # __checkpoint_upload_mode_no_upload_end__
 
 
-# __checkpoint_upload_function_start__
+# __checkpoint_upload_fn_start__
 
 from torch.distributed.checkpoint.state_dict_saver import async_save
 from s3torchconnector.dcp import S3StorageWriter
@@ -542,11 +542,11 @@ def wait_async_save(checkpoint, checkpoint_dir_name):
             metrics=metrics,
             checkpoint=checkpoint,
             checkpoint_upload_mode=train.CheckpointUploadMode.ASYNC,
-            checkpoint_upload_function=wait_async_save,
+            checkpoint_upload_fn=wait_async_save,
         )
 
 
-# __checkpoint_upload_function_end__
+# __checkpoint_upload_fn_end__
 
 # __get_all_reported_checkpoints_example_start__
 

doc/source/train/user-guides/checkpoints.rst

@@ -312,8 +312,8 @@ you have the following options:
 
         .. literalinclude:: ../doc_code/checkpoints.py
             :language: python
-            :start-after: __checkpoint_upload_function_start__
-            :end-before: __checkpoint_upload_function_end__
+            :start-after: __checkpoint_upload_fn_start__
+            :end-before: __checkpoint_upload_fn_end__
 
         .. warning::