Skip to content

GitLab

Unverified Commit 722bf7ef authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix missing `()` in some usage of `is_flaky` (#20749) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 9bafedc0
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/models/donut/test_feature_extraction_donut.py
View file @ 722bf7ef
...@@ -106,7 +106,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test ...@@ -106,7 +106,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test
def test_batch_feature(self): def test_batch_feature(self):
pass pass
@is_flaky @is_flaky()
def test_call_pil(self): def test_call_pil(self):
# Initialize feature_extractor # Initialize feature_extractor
feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) feature_extractor = self.feature_extraction_class(**self.feat_extract_dict)
...@@ -139,7 +139,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test ...@@ -139,7 +139,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test
), ),
) )
@is_flaky @is_flaky()
def test_call_numpy(self): def test_call_numpy(self):
# Initialize feature_extractor # Initialize feature_extractor
feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) feature_extractor = self.feature_extraction_class(**self.feat_extract_dict)
...@@ -172,7 +172,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test ...@@ -172,7 +172,7 @@ class DonutFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest.Test
), ),
) )
@is_flaky @is_flaky()
def test_call_pytorch(self): def test_call_pytorch(self):
# Initialize feature_extractor # Initialize feature_extractor
feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) feature_extractor = self.feature_extraction_class(**self.feat_extract_dict)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment