Skip to content

Commit 5152c2a

Browse files
committed
fix pylint
1 parent 1732ad2 commit 5152c2a

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

tests/test_backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1540,6 +1540,7 @@ def test_sparse_softmax_cross_entropy_with_logits_large_class(self):
15401540

15411541
self._run_test_case([_OUTPUT], {_INPUT: label_val, _INPUT1: logits_val})
15421542

1543+
@unittest.skipIf(BACKEND in ["onnxruntime"], "onnxruntime Slice did not supported BOOL.")
15431544
def test_matrix_band_part(self):
15441545
input_val = np.random.randint(0, 666, (10, 15)).astype(np.int32)
15451546
input_x = tf.placeholder(dtype=tf.int32, shape=[None, None], name=_TFINPUT)
@@ -1549,6 +1550,7 @@ def test_matrix_band_part(self):
15491550
_ = tf.identity(res1, name=_TFOUTPUT1)
15501551
self._run_test_case([_OUTPUT, _OUTPUT1], {_INPUT: input_val})
15511552

1553+
@unittest.skipIf(BACKEND in ["onnxruntime"], "onnxruntime Slice did not supported BOOL.")
15521554
def test_matrix_band_part_2(self):
15531555
input_val = np.random.randint(0, 666, (1, 1)).astype(np.int32)
15541556
input_x = tf.placeholder(dtype=tf.int32, shape=[None, None], name=_TFINPUT)

tf2onnx/tfonnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2263,7 +2263,7 @@ def rewrite_incomplete_type_support(g, ops, impacted_ops):
22632263
input_name = op.input[i]
22642264
dtype = g.get_dtype(input_name)
22652265
if dtype is None:
2266-
log.warning("For %s (type is %s)' input: %s, its dtype should not be None if we need add Cast for it.",
2266+
log.warning("Adding Cast for op %s (type is %s)' input: %s, dtype should not be None",
22672267
op.name, op.type, input_name)
22682268

22692269
if dtype != onnx_pb.TensorProto.FLOAT:

0 commit comments

Comments
 (0)