fix(docs): fix traced module docs warning

GitOrigin-RevId: a5b790c7b5

imperative/python/megengine/traced_module/__init__.py

@@ -21,10 +21,10 @@ from .traced_module import (
 _register_all_builtin_module()
 set_cpp_apply_module_trace(cpp_apply_module_trace)
 
-__all__ = {
+__all__ = [
     "register_as_builtin",
     "trace_module",
     "wrap",
     "TracedModule",
     "optimize",
-}
+]

imperative/python/megengine/traced_module/node.py

@@ -222,7 +222,7 @@ class TensorNode(Node):
 
     def __init__(
         self,
-        expr: "Expr",
+        expr,
         name: str = None,
         qualname: str = None,
         shape: Tuple[int] = None,