fix torch.distributed.rpc example incorrect usage (#112367)

Fixes #112366
Pull Request resolved: #112367
Approved by: https://github.com/H-Huang

torch/distributed/rpc/api.py

@@ -612,8 +612,8 @@ def remote(to, func, args=None, kwargs=None, timeout=UNSET_RPC_TIMEOUT):
 
         >>> # On both workers:
         >>> @torch.jit.script
-        >>> def my_script_add(t1, t2):
-        >>>    return torch.add(t1, t2)
+        >>> def my_script_add(tensor: torch.Tensor, scalar: int):
+        >>>    return torch.add(tensor, scalar)
 
         >>> # On worker 0:
         >>> import torch.distributed.rpc as rpc
@@ -793,8 +793,8 @@ def rpc_sync(to, func, args=None, kwargs=None, timeout: float = UNSET_RPC_TIMEOU
 
         >>> # On both workers:
         >>> @torch.jit.script
-        >>> def my_script_add(t1, t2):
-        >>>    return torch.add(t1, t2)
+        >>> def my_script_add(tensor: torch.Tensor, scalar: int):
+        >>>    return torch.add(tensor, scalar)
 
         >>> # On worker 0:
         >>> import torch.distributed.rpc as rpc
@@ -887,8 +887,8 @@ def rpc_async(to, func, args=None, kwargs=None, timeout=UNSET_RPC_TIMEOUT):
 
         >>> # On both workers:
         >>> @torch.jit.script
-        >>> def my_script_add(t1, t2):
-        >>>    return torch.add(t1, t2)
+        >>> def my_script_add(tensor: torch.Tensor, scalar: int):
+        >>>    return torch.add(tensor, scalar)
 
         >>> # On worker 0:
         >>> import torch.distributed.rpc as rpc