Skip to content

Commit 4eae48f

Browse files
committed
Set skip tf2 flag correctly
1 parent 941d4cd commit 4eae48f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/test_backend.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1917,7 +1917,8 @@ def graph_validator(g):
19171917

19181918
self._run_test_case(func_fusedbn, [_OUTPUT], {_INPUT: x_val}, rtol=1e-05, graph_validator=graph_validator)
19191919

1920-
@check_tf_min_version("1.15", "skip_tf2")
1920+
@check_tf_min_version("1.15")
1921+
@skip_tf2()
19211922
@check_opset_min_version(10, "quantize_and_dequantize")
19221923
def test_qdq_unsigned_input(self):
19231924
x_shape = [3, 3, 2]
@@ -1927,7 +1928,8 @@ def func(x):
19271928
return tf.identity(x_, name=_TFOUTPUT)
19281929
_ = self._run_test_case(func, [_OUTPUT], {_INPUT: x_val})
19291930

1930-
@check_tf_min_version("1.15", "skip_tf2")
1931+
@check_tf_min_version("1.15")
1932+
@skip_tf2()
19311933
@check_opset_min_version(10, "quantize_and_dequantize")
19321934
def test_qdq_signed_input(self):
19331935
x_shape = [3, 3, 2]

0 commit comments

Comments
 (0)