pylint

Author: pengwa

tests/common.py

@@ -24,7 +24,7 @@ class TestConfig(object):
     def __init__(self):
         self.platform = sys.platform
         self.tf_version = self._get_tf_version()
-        self.opset = int(os.environ.get("TF2ONNX_TEST_OPSET", 9))
+        self.opset = int(os.environ.get("TF2ONNX_TEST_OPSET", 7))
         self.target = os.environ.get("TF2ONNX_TEST_TARGET", ",".join(DEFAULT_TARGET)).split(',')
         self.backend = os.environ.get("TF2ONNX_TEST_BACKEND", "onnxruntime")
         self.backend_version = self._get_backend_version()

tf2onnx/rewriter/cond_rewriter.py

@@ -12,7 +12,6 @@
 from collections import OrderedDict
 from enum import Enum
 from tf2onnx import utils
-from tf2onnx.utils import TensorValueInfo
 
 logging.basicConfig(level=logging.INFO)
 log = logging.getLogger("tf2onnx.rewriter.cond_rewriter_base")

tf2onnx/tfonnx.py

@@ -199,7 +199,6 @@ def broadcast_op(ctx, node, name, args):
             node.input[1] = tmp
     else:
         node.set_attr("broadcast", 0)
-        return node
 
 
 def broadcast_op7(ctx, node, name, args):
@@ -221,7 +220,6 @@ def broadcast_op7(ctx, node, name, args):
             tmp = node.input[0]
             node.input[0] = node.input[1]
             node.input[1] = tmp
-    return node
 
 
 def arg_minmax_op(ctx, node, name, args):
@@ -682,7 +680,7 @@ def biasadd_op7(ctx, node, name, args):
     # T output = BiasAddV1(T value, T bias)
     # According TF bias_add definition, the input dim is always only 1.
     node.type = "Add"
-    node = broadcast_op7(ctx, node, name, args)
+    broadcast_op7(ctx, node, name, args)
 
     # on NHWC, bias will broadcast from largest dim, which is default onnx Add op broadcast behavior.
     if not node.is_nhwc():