From a8941af0c654f6de2d6cb5b6057b568f02c1a432 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 29 Nov 2021 17:49:23 +0800
Subject: [PATCH] fix(docs): fix traced module docs warning

GitOrigin-RevId: a5b790c7b527c700a594109d6d0219cd8b359e68
---
 imperative/python/megengine/traced_module/__init__.py | 4 ++--
 imperative/python/megengine/traced_module/node.py     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/imperative/python/megengine/traced_module/__init__.py b/imperative/python/megengine/traced_module/__init__.py
index 4256f5e03..c906b8790 100644
--- a/imperative/python/megengine/traced_module/__init__.py
+++ b/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",
-}
+]
diff --git a/imperative/python/megengine/traced_module/node.py b/imperative/python/megengine/traced_module/node.py
index 96091b36b..4bfff4afc 100644
--- a/imperative/python/megengine/traced_module/node.py
+++ b/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,
-- 
GitLab