clean up flatbuffers import

uTensor · Nov 29, 2019 · 36f1fe3 · 36f1fe3
1 parent c034f7d
commit 36f1fe3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/tests/tflm/tflite_export/test_write.py b/tests/tflm/tflite_export/test_write.py
@@ -5,9 +5,8 @@
 from utensor_cgen.matcher import uTensorGraphMatcher
 from utensor_cgen.utils import prune_graph, topologic_order_graph
 from utensor_cgen.transformer import TFLiteExporter
-import flatbuffers
+import utensor_cgen.third_party.flatbuffers as flatbuffers
 import utensor_cgen.third_party.tflite as tflite
-from utensor_cgen.third_party.tflite import *
 from utensor_cgen.third_party.tflite.BuiltinOperator import BuiltinOperator
 from utensor_cgen.third_party.tflite.Model import Model
 from utensor_cgen.third_party.tflite.BuiltinOptions import BuiltinOptions

diff --git a/utensor_cgen/transformer/tflite_exporter.py b/utensor_cgen/transformer/tflite_exporter.py
@@ -9,7 +9,6 @@
 from copy import deepcopy
 
 from utensor_cgen.ir import OperationInfo, uTensorGraph
-#import flatbuffers
 import utensor_cgen.third_party.flatbuffers as flatbuffers
 import utensor_cgen.third_party.tflite as tflite
 from utensor_cgen.third_party.tflite import *