Skip to content

Commit b51b273

Browse files
[internal] Reorder @pytest.mark.large decorator to fix CI (#2410)
1 parent 73ca767 commit b51b273

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_hub/src/models/task_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,6 @@ def test_summary_without_preprocessor(self):
108108
model.summary(print_fn=lambda x, line_break=False: summary.append(x))
109109
self.assertNotRegex("\n".join(summary), "Preprocessor:")
110110

111-
@pytest.mark.large
112111
@parameterized.named_parameters(
113112
{
114113
"testcase_name": "load_with_quantized_weights",
@@ -123,6 +122,7 @@ def test_summary_without_preprocessor(self):
123122
"expected_dtype": "float32",
124123
},
125124
)
125+
@pytest.mark.large
126126
def test_quantized_preset_loading_and_saving(
127127
self, load_weights, dtype_override, expected_dtype
128128
):

0 commit comments

Comments
 (0)