From 9c4b4071f56ba41b3d336c1565fb784b7fd2aed7 Mon Sep 17 00:00:00 2001 From: guschmue Date: Mon, 1 Feb 2021 11:04:42 -0800 Subject: [PATCH] test_unsorted_segment_ops needs tf-2.3 or better Signed-off-by: guschmue --- tests/test_backend.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_backend.py b/tests/test_backend.py index 66d613c7d..b2090f68a 100644 --- a/tests/test_backend.py +++ b/tests/test_backend.py @@ -1532,6 +1532,7 @@ def func(data, indices, segments): self._run_test_case(func, [_OUTPUT], {_INPUT: data_val, _INPUT1: indices_val, _INPUT2: segs_val}) @check_opset_min_version(9, "OneHot") + @check_tf_min_version("2.3", "needs tf 2.3") def test_unsorted_segment_ops(self): tf_ops = [ tf.math.unsorted_segment_max,