fix(traced_module/doc): fix warning for traced_module docstring

GitOrigin-RevId: 5d2d047d2fb226d629ab5e18d9d5221d27289c6a

imperative/python/megengine/traced_module/expr.py

@@ -159,13 +159,13 @@ class Expr:
 
     @property
     def kwargs(self):
-        r"""Get the the keyword arguments of the operation corresponding to this Expr."""
+        r"""Get the keyword arguments of the operation corresponding to this Expr."""
         _, kwargs = self.unflatten_args(self.inputs)
         return kwargs
 
     @property
     def args(self):
-        r"""Get the the positional arguments of the operation corresponding to this Expr."""
+        r"""Get the positional arguments of the operation corresponding to this Expr."""
         args, _ = self.unflatten_args(self.inputs)
         return args
 

imperative/python/megengine/traced_module/node.py

@@ -33,7 +33,7 @@ class Node:
     _orig_name = None  # type: str
     _format_spec = ""  # type: str
 
-    def __init__(self, expr: "Expr", name: str, orig_name: str):
+    def __init__(self, expr, name: str, orig_name: str):
         self.expr = expr
         self.users = []  # List[Expr]
         self._id = Node.__total_id
@@ -120,7 +120,7 @@ class ModuleNode(Node):
     r"""The type of the Module correspending to the ModuleNode."""
     _owner = None  # type: weakref.ReferenceType
 
-    def __init__(self, expr: "Expr", name: str = None, orig_name: str = None):
+    def __init__(self, expr, name: str = None, orig_name: str = None):
         super().__init__(expr, name, orig_name)
 
     def __getstate__(self):
@@ -136,9 +136,6 @@ class ModuleNode(Node):
     @property
     def owner(self):
         r"""Get the ``Module`` corresponding to this ``ModuleNode``.
-
-        Returns:
-            An :calss:`~.Module`.
         """
         if self._owner:
             return self._owner()
@@ -196,7 +193,7 @@ class TensorNode(Node):
 
     @property
     def qparams(self):
-        r"""Get the :calss:`QParams` of this Node."""
+        r"""Get the :class:`QParams` of this Node."""
         return self._qparams
 
     @qparams.setter
@@ -210,11 +207,7 @@ class TensorNode(Node):
 
     @value.setter
     def value(self, value):
-        r"""Bind a Tensor to this Node.
-
-        Args:
-            value: A :class:`Tensor`.
-        """
+        r"""Bind a :class:`Tensor` to this Node."""
         if isinstance(value, RawTensor) and NodeMixin.get(value, None) is not None:
             setattr(value, "_NodeMixin__node", None)
         self._value = value

imperative/python/megengine/traced_module/pytree.py

@@ -150,8 +150,8 @@ def tree_flatten(
     is_leaf: Callable = _is_leaf,
     is_const_leaf: Callable = _is_const_leaf,
 ):
-    r"""Flattens a object into a list of values and a :calss:`TreeDef` that can be used
-    to reconstruct the object.
+    r"""Flattens a pytree into a list of values and a :class:`TreeDef` that can be used
+    to reconstruct the pytree.
     """
     if type(values) not in SUPPORTED_TYPE:
         assert is_leaf(values), values
@@ -188,7 +188,7 @@ class TreeDef:
         self.num_leaves = sum(ch.num_leaves for ch in children_defs)
 
     def unflatten(self, leaves):
-        r"""Given a list of values and a ``TreeDef``, builds a object.
+        r"""Given a list of values and a ``TreeDef``, builds a pytree.
         This is the inverse operation of ``tree_flatten``.
         """
         assert len(leaves) == self.num_leaves

imperative/python/megengine/traced_module/traced_module.py

@@ -453,7 +453,7 @@ class InternalGraph:
         r"""Get the list of output Nodes of this graph.
 
         Returns:
-            A list of Node.
+            A list of ``Node``.
         """
         return self._outputs
 
@@ -1937,7 +1937,7 @@ class TracedModule(Module):
 
     @property
     def graph(self) -> InternalGraph:
-        """Return the ``InternalGraph`` of this ``TracedModule``
+        """Return the ``InternalGraph`` of this ``TracedModule``.
         """
         if self._is_top:
             self._update_ref()