pylint

Author: sdpython

tests/backend_test_base.py

@@ -21,6 +21,7 @@
 import tensorflow as tf
 from tensorflow.python.ops import variables as variables_lib
 from common import get_test_config
+import onnx
 from tf2onnx import utils
 from tf2onnx.tfonnx import process_tf_graph
 from tf2onnx import optimizer
@@ -72,14 +73,12 @@ def run_onnxruntime(self, model_path, inputs, output_names):
         # opt.log_severity_level = 0
         # opt.log_verbosity_level = 255
         # opt.enable_profiling = True
-        import onnx
         with open(model_path, 'rb') as f:
             onx = onnx.load(f)
         print(onx)
         try:
             m = rt.InferenceSession(model_path, opt)
         except Fail as e:
-            import onnx
             with open(model_path, 'rb') as f:
                 onx = onnx.load(f)
             raise AssertionError("Unable to load model '{}'\n{}".format(model_path, onx)) from e

tf2onnx/graph.py

@@ -9,6 +9,8 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
+import traceback
+import sys
 import collections
 import copy
 import logging
@@ -1305,8 +1307,6 @@ def replace_all_inputs(self, ops, old_input, new_input, debug=False, keep_ops=Fa
         try:
             assert False
         except AssertionError:
-            import traceback
-            import sys
             tb = traceback.extract_stack()
             print()
             for line in tb: