remove pylint warnings

onnx · Sep 27, 2019 · a8a28ee · a8a28ee
1 parent f084b77
commit a8a28ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/tests/test_backend.py b/tests/test_backend.py
@@ -2780,7 +2780,6 @@ def test_determinant(self):
 
     @check_opset_min_version(11, "BitShift")
     def test_bitshift_left(self):
-        info = np.iinfo(np.int32)
         x_val = np.array([16, 4, 1], dtype=np.int32)
         y_val = np.array([1, 2, 3], dtype=np.int32)
         x = tf.placeholder(tf.int32, x_val.shape, name=_TFINPUT)

diff --git a/tf2onnx/onnx_opset/math.py b/tf2onnx/onnx_opset/math.py
@@ -504,7 +504,7 @@ def version_11(cls, ctx, node, **kwargs):
 
 
 @tf_op("LeftShift", onnx_op="BitShift")
-class BitShift:
+class BitShiftLeft:
     @classmethod
     def version_11(cls, ctx, node, **kwargs):
         shapes = node.output_shapes
@@ -516,7 +516,7 @@ def version_11(cls, ctx, node, **kwargs):
 
 
 @tf_op("RightShift", onnx_op="BitShift")
-class BitShift:
+class BitShiftRight:
     @classmethod
     def version_11(cls, ctx, node, **kwargs):
         shapes = node.output_shapes