Mark preset tests as large (keras-team#1942)

* fix tests * fix test * Update preset_utils_test.py
ushareng · Oct 24, 2024 · 500a2e9 · 500a2e9
1 parent 150fae2
commit 500a2e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras_hub/src/utils/preset_utils_test.py b/keras_hub/src/utils/preset_utils_test.py
@@ -18,6 +18,7 @@
 
 
 class PresetUtilsTest(TestCase):
+    @pytest.mark.large
     def test_preset_errors(self):
         with self.assertRaisesRegex(ValueError, "must be a string"):
             AlbertTextClassifier.from_preset(AlbertTextClassifier)
@@ -34,6 +35,7 @@ def test_preset_errors(self):
         with self.assertRaisesRegex(ValueError, "class keras_hub>BortBackbone"):
             BertBackbone.from_preset(preset_dir)
 
+    @pytest.mark.large
     def test_upload_empty_preset(self):
         temp_dir = self.get_temp_dir()
         empty_preset = os.path.join(temp_dir, "empty")