From 722bf7efcce72e60412f75d6775af7b03041d8c8 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 14 Dec 2022 11:37:29 +0100 Subject: [PATCH] Fix missing `()` in some usage of `is_flaky` (#20749) Co-authored-by: ydshieh --- tests/models/donut/test_feature_extraction_donut.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/models/donut/test_feature_extraction_donut.py b/tests/models/donut/test_feature_extraction_donut.py index f159c07085..e97c32f6c9 100644 --- a/tests/models/donut/test_feature_extraction_donut.py +++ b/tests/models/donut/test_feature_extraction_donut.py @@ -106,7 +106,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test def test_batch_feature(self): pass - @is_flaky + @is_flaky() def test_call_pil(self): # Initialize feature_extractor feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) @@ -139,7 +139,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test ), ) - @is_flaky + @is_flaky() def test_call_numpy(self): # Initialize feature_extractor feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) @@ -172,7 +172,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test ), ) - @is_flaky + @is_flaky() def test_call_pytorch(self): # Initialize feature_extractor feature_extractor = self.feature_extraction_class(**self.feat_extract_dict)