Skip to content

Commit 7b8991c

Browse files
committed
disable onehot unit test
it's ok in local env, while in unit test the target info is not given so disable them for now
1 parent 2594eca commit 7b8991c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/test_backend.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1118,6 +1118,7 @@ def test_sign(self):
11181118
self._run_test_case([_OUTPUT], {_INPUT: x_val})
11191119
tf.reset_default_graph()
11201120

1121+
@check_target("rs6", "onehot")
11211122
def test_onehot0(self):
11221123
x_val = np.array([0, 1, 2], dtype=np.int32)
11231124
depth = 5
@@ -1138,6 +1139,7 @@ def test_onehot1(self):
11381139
_ = tf.identity(x_, name=_TFOUTPUT)
11391140
self._run_test_case([_OUTPUT], {_INPUT: x_val})
11401141

1142+
@check_target("rs6", "onehot")
11411143
def test_onehot2(self):
11421144
for axis in [-1, 0, 1]:
11431145
tf.reset_default_graph()
@@ -1148,6 +1150,7 @@ def test_onehot2(self):
11481150
_ = tf.identity(x_, name=_TFOUTPUT)
11491151
self._run_test_case([_OUTPUT], {_INPUT: x_val})
11501152

1153+
@check_target("rs6", "onehot")
11511154
@check_opset_min_version(9, "onehot")
11521155
def test_onehot3(self):
11531156
# rank 1

0 commit comments

Comments
 (0)