From 954b9d3149e1ffcae1fd9ea726b9628b30b4d515 Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Fri, 4 Nov 2022 05:02:48 +0300 Subject: [PATCH] [TF FE][Tests] Add layer tests for SegmentSum and DynamicPartition in the pre-commit (#13812) After fixes for TopK and arithmetic operations for integer tensors, we are ready to switch these tests in the pre-commit Signed-off-by: Kazantsev, Roman Signed-off-by: Kazantsev, Roman --- tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py | 2 +- .../tensorflow_tests/test_tf_ParallelDynamicStitch.py | 2 +- tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py b/tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py index 32ff58c262c..f3caa40576d 100644 --- a/tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py +++ b/tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py @@ -42,7 +42,7 @@ class TestDynamicPartition(CommonTFLayerTest): ] @pytest.mark.parametrize("params", test_data_basic) - @pytest.mark.nightly + @pytest.mark.precommit_tf_fe def test_dynamic_partition_basic(self, params, ie_device, precision, ir_version, temp_dir, use_new_frontend, use_old_api): if not use_new_frontend: diff --git a/tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py b/tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py index eb940b6d96e..901eb0236c9 100644 --- a/tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py +++ b/tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py @@ -88,4 +88,4 @@ class TestParallelDynamicStitch(CommonTFLayerTest): pytest.skip("DynamicStitch operation is not supported via legacy frontend.") self._test(*self.create_parallel_dynamic_stitch_net(**params), ie_device, precision, ir_version, temp_dir=temp_dir, - use_new_frontend=use_new_frontend, use_old_api=use_old_api) \ No newline at end of file + use_new_frontend=use_new_frontend, use_old_api=use_old_api) diff --git a/tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py b/tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py index 14dabde255d..b40c09ddc4b 100644 --- a/tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py +++ b/tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py @@ -39,7 +39,7 @@ class TestSegmentSum(CommonTFLayerTest): ] @pytest.mark.parametrize("params", test_data_basic) - @pytest.mark.nightly + @pytest.mark.precommit_tf_fe def test_segment_sum_basic(self, params, ie_device, precision, ir_version, temp_dir, use_new_frontend, use_old_api): if not use_new_frontend: