We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8a943da commit 12807baCopy full SHA for 12807ba
keras_hub/src/utils/preset_utils_test.py
@@ -18,6 +18,7 @@
18
19
20
class PresetUtilsTest(TestCase):
21
+ @pytest.mark.large
22
def test_preset_errors(self):
23
with self.assertRaisesRegex(ValueError, "must be a string"):
24
AlbertTextClassifier.from_preset(AlbertTextClassifier)
@@ -34,6 +35,7 @@ def test_preset_errors(self):
34
35
with self.assertRaisesRegex(ValueError, "class keras_hub>BortBackbone"):
36
BertBackbone.from_preset(preset_dir)
37
38
39
def test_upload_empty_preset(self):
40
temp_dir = self.get_temp_dir()
41
empty_preset = os.path.join(temp_dir, "empty")
0 commit comments