fix some

PaddlePaddle · Oct 9, 2023 · 2a44dfb · 2a44dfb
1 parent 84810bf
commit 2a44dfb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/paddle/distributed/sharding/group_sharded.py b/python/paddle/distributed/sharding/group_sharded.py
@@ -77,7 +77,7 @@ def group_sharded_parallel(
     Examples:
         .. code-block:: python
 
-            >>> # doctest: +REQUIRES(env:distributed)
+            >>> # doctest: +REQUIRES(env:DISTRIBUTED)
             >>> import paddle
             >>> from paddle.nn import Linear
             >>> from paddle.distributed import fleet
@@ -196,7 +196,7 @@ def save_group_sharded_model(model, output, optimizer=None):
     Examples:
         .. code-block:: python
 
-            >>> # doctest: +REQUIRES(env:distributed)
+            >>> # doctest: +REQUIRES(env:DISTRIBUTED)
             >>> import paddle
             >>> from paddle.nn import Linear
             >>> from paddle.distributed import fleet

diff --git a/python/paddle/distributed/transpiler/distribute_transpiler.py b/python/paddle/distributed/transpiler/distribute_transpiler.py
@@ -316,7 +316,7 @@ class DistributeTranspiler:
             ...         pserver_program = t.get_pserver_program(current_endpoint)
             ...         pserver_startup_program = t.get_startup_program(current_endpoint,
             ...                                                     pserver_program)
-            >>> elif role == "TRAINER":
+            ... elif role == "TRAINER":
             ...         trainer_program = t.get_trainer_program()
 
             >>> # for nccl2 mode