Skip to content

Commit 70f40a4

Browse files
committed
Fix pylint errors and unit tests
- fixed pylint errors - fixed some broken for-loop logic
1 parent fc75c26 commit 70f40a4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_stacked_lstm.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from tensorflow.python.ops import init_ops
77
from backend_test_base import Tf2OnnxBackendTestBase
8-
from common import unittest_main, check_lstm_count
8+
from common import unittest_main, check_lstm_count, check_opset_after_tf_version, skip_tf2
99

1010
from tf2onnx.tf_loader import is_tf2
1111

@@ -25,6 +25,8 @@
2525

2626

2727
class LSTMLayeredTests(Tf2OnnxBackendTestBase):
28+
@check_opset_after_tf_version("1.15", 8, "might need Scan")
29+
@skip_tf2()
2830
def test_layered_lstm(self):
2931
units = 5
3032
batch_size = 6

0 commit comments

Comments
 (0)