Skip to content

Commit 3b669e9

Browse files
committed
re-enable some tests
1 parent 9aad343 commit 3b669e9

File tree

5 files changed

+1
-14
lines changed

5 files changed

+1
-14
lines changed

tests/test_backend.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -191,8 +191,6 @@ def test_maxpool(self):
191191
self.logger.debug(str(p))
192192
self._run_test_case([_OUTPUT], {_INPUT: x_val})
193193

194-
@unittest.skipIf(get_test_config().is_onnxruntime_backend and get_test_config().backend_version == "0.2.1",
195-
"onnxruntime bug")
196194
@check_onnxruntime_incompatibility("AveragePool")
197195
def test_avgpool(self):
198196
for tf_shape in ["known", "unknown"]:
@@ -1195,7 +1193,7 @@ def test_randomuniform(self):
11951193
# since results are random, compare the shapes only
11961194
self._run_test_case([_OUTPUT], {}, check_value=False, check_shape=True)
11971195

1198-
@unittest.skip("")
1196+
@unittest.skip("TF RandomUniformInt is not supported")
11991197
def test_randomuniform_int(self):
12001198
shape = tf.constant([2, 3], name="shape")
12011199
x_ = tf.random_uniform(shape, name="rand", dtype=tf.int32, maxval=10)

tests/test_gru.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from __future__ import print_function
99
from __future__ import unicode_literals
1010

11-
import unittest
1211
import numpy as np
1312
import tensorflow as tf
1413

@@ -209,7 +208,6 @@ def test_single_dynamic_gru_ch_zero_state_initializer(self):
209208
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=1e-03, atol=1e-06,
210209
graph_validator=lambda g: check_gru_count(g, 1))
211210

212-
@unittest.skip("FIXME: disable for now for accuracy problem")
213211
def test_single_dynamic_gru_random_weights(self):
214212
hidden_size = 5
215213
batch_size = 1
@@ -238,7 +236,6 @@ def test_single_dynamic_gru_random_weights(self):
238236
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, 0.0001,
239237
graph_validator=lambda g: check_gru_count(g, 1))
240238

241-
@unittest.skip("FIXME: disable for now for accuracy problem")
242239
def test_single_dynamic_gru_random_weights2(self):
243240
hidden_size = 128
244241
batch_size = 1

tests/test_grublock.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from __future__ import print_function
99
from __future__ import unicode_literals
1010

11-
import unittest
1211
import numpy as np
1312
import tensorflow as tf
1413

@@ -194,7 +193,6 @@ def test_single_dynamic_gru_ch_zero_state_initializer(self):
194193
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=1e-03, atol=1e-06,
195194
graph_validator=lambda g: check_gru_count(g, 1))
196195

197-
@unittest.skip("FIXME: disable for now for accuracy problem")
198196
def test_single_dynamic_gru_random_weights(self):
199197
hidden_size = 5
200198
batch_size = 1
@@ -220,7 +218,6 @@ def test_single_dynamic_gru_random_weights(self):
220218
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, 0.0001,
221219
graph_validator=lambda g: check_gru_count(g, 1))
222220

223-
@unittest.skip("FIXME: disable for now for accuracy problem")
224221
def test_single_dynamic_gru_random_weights2(self):
225222
hidden_size = 128
226223
batch_size = 1

tests/test_loops.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from __future__ import print_function
99
from __future__ import unicode_literals
1010

11-
import unittest
1211
import numpy as np
1312
import tensorflow as tf
1413

@@ -103,7 +102,6 @@ def b(i, res, res2):
103102
output_names_with_port = ["i:0", "x:0", "y:0"]
104103
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=1e-06)
105104

106-
@unittest.skip("bug in onnxruntime")
107105
def test_while_loop_with_ta_read_reference_outer_input_directly(self):
108106
i = tf.placeholder(tf.int32, (), name="input_1")
109107
inputs = tf.placeholder(tf.float32, (10,), name="input_2")

tests/test_lstm.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from __future__ import print_function
99
from __future__ import unicode_literals
1010

11-
import unittest
1211
import numpy as np
1312
import tensorflow as tf
1413

@@ -268,7 +267,6 @@ def test_single_dynamic_lstm_consume_one_of_ch_tuple(self):
268267
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=1e-06,
269268
graph_validator=lambda g: check_lstm_count(g, 1))
270269

271-
@unittest.skip("FIXME: disable for now for accuracy problem")
272270
def test_single_dynamic_lstm_random_weights(self, state_is_tuple=True):
273271
hidden_size = 5
274272
batch_size = 6
@@ -298,7 +296,6 @@ def test_single_dynamic_lstm_random_weights(self, state_is_tuple=True):
298296
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=0.0001,
299297
graph_validator=lambda g: check_lstm_count(g, 1))
300298

301-
@unittest.skip("FIXME: disable for now for accuracy problem")
302299
def test_single_dynamic_lstm_random_weights2(self, state_is_tuple=True):
303300
hidden_size = 128
304301
batch_size = 1

0 commit comments

Comments
 (0)