From 1df92ed86802b99e3b911c6f1f35f49ab58ce388 Mon Sep 17 00:00:00 2001
From: liangtianshu <liangtianshu@huawei.com>
Date: Wed, 27 Jan 2021 17:30:46 +0800
Subject: [PATCH] fix a function has 51 lines

---
 .../graph_based_converter/generator/generator.py             | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/mindinsight/mindconverter/graph_based_converter/generator/generator.py b/mindinsight/mindconverter/graph_based_converter/generator/generator.py
index 154f6860..049b0103 100644
--- a/mindinsight/mindconverter/graph_based_converter/generator/generator.py
+++ b/mindinsight/mindconverter/graph_based_converter/generator/generator.py
@@ -105,7 +105,7 @@ class CodeStruct:
                 init_lines += init_str
                 cons_lines += cons_str
 
-            elif isinstance(struct, ModuleStruct):
+            else: # is ModuleStruct
                 # check if this instance generated CodeStruct
                 if GlobalContext().code_structs.get(struct.pattern_id) is None:
                     CodeStruct(struct, repeated_submodules)
@@ -115,9 +115,6 @@ class CodeStruct:
                 init_lines.append(f"{SECOND_LEVEL_INDENT}{' = '.join(code_line_init)}")
                 cons_lines.append(f"{SECOND_LEVEL_INDENT}{' = '.join(code_line_construct)}")
 
-            else:
-                raise TypeError("Unable to generate code from args are not ModuleStruct or NodeStruct.")
-
         # define header of init block
         self.new_line = f"{FIRST_LEVEL_INDENT}def __init__({', '.join(module_def_args)}):"
         self.new_line = f"{SECOND_LEVEL_INDENT}super({class_name}, self).__init__()"