Skip to content

Commit 500a2e9

Browse files
divyashreepathihalliushareng
authored andcommitted
Mark preset tests as large (keras-team#1942)
* fix tests * fix test * Update preset_utils_test.py
1 parent 150fae2 commit 500a2e9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras_hub/src/utils/preset_utils_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919

2020
class PresetUtilsTest(TestCase):
21+
@pytest.mark.large
2122
def test_preset_errors(self):
2223
with self.assertRaisesRegex(ValueError, "must be a string"):
2324
AlbertTextClassifier.from_preset(AlbertTextClassifier)
@@ -34,6 +35,7 @@ def test_preset_errors(self):
3435
with self.assertRaisesRegex(ValueError, "class keras_hub>BortBackbone"):
3536
BertBackbone.from_preset(preset_dir)
3637

38+
@pytest.mark.large
3739
def test_upload_empty_preset(self):
3840
temp_dir = self.get_temp_dir()
3941
empty_preset = os.path.join(temp_dir, "empty")

0 commit comments

Comments
 (0)