Mark preset tests as large (#1942)

* fix tests * fix test * Update preset_utils_test.py
keras-team · Oct 21, 2024 · 12807ba · 12807ba
1 parent 8a943da
commit 12807ba
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")