[Typing] 修复标注类型 ReduceOp to _ReduceOp (#66884)

megemini · web-flow · commit 71ab4274502f · 2024-08-01T18:33:54.000+08:00
diff --git a/python/paddle/distributed/communication/all_reduce.py b/python/paddle/distributed/communication/all_reduce.py
@@ -23,11 +23,12 @@
     from paddle import Tensor
     from paddle.base.core import task
     from paddle.distributed.communication.group import Group
+    from paddle.distributed.communication.reduce import _ReduceOp
 
 
 def all_reduce(
     tensor: Tensor,
-    op: ReduceOp = ReduceOp.SUM,
+    op: _ReduceOp = ReduceOp.SUM,
     group: Group | None = None,
     sync_op: bool = True,
 ) -> task:
