Skip to content
This repository was archived by the owner on Oct 13, 2021. It is now read-only.

Commit 9cc6446

Browse files
committed
Excluding TF2 from tests for RNN initial states
1 parent ffb7041 commit 9cc6446

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_layers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1574,6 +1574,7 @@ def test_Bidirectional_with_bias(self):
15741574
onnx_model = keras2onnx.convert_keras(model, model.name)
15751575
self.assertTrue(run_onnx_runtime(onnx_model.graph.name, onnx_model, x, expected, self.model_files))
15761576

1577+
@unittest.skipIf(is_tf2 and is_tf_keras, 'TODO')
15771578
def test_Bidirectional_with_initial_states(self):
15781579
for rnn_class in [SimpleRNN, GRU, LSTM]:
15791580
input1 = Input(shape=(None, 5))
@@ -1613,6 +1614,7 @@ def test_Bidirectional_seqlen_none(self):
16131614
expected = model.predict(x)
16141615
self.assertTrue(run_onnx_runtime(onnx_model.graph.name, onnx_model, x, expected, self.model_files))
16151616

1617+
@unittest.skipIf(is_tf2, 'TODO')
16161618
def test_rnn_state_passing(self):
16171619
for rnn_class in [SimpleRNN, GRU, LSTM]:
16181620
input1 = Input(shape=(None, 5))

0 commit comments

Comments
 (0)