Mark slow tests as slow

2022-03-03 11:03:24 -05:00
parent 3822e4a563
commit 7c45fe747f
2 changed files with 6 additions and 0 deletions
--- a/tests/maskformer/test_modeling_maskformer.py
+++ b/tests/maskformer/test_modeling_maskformer.py
@@ -161,6 +161,7 @@ class MaskFormerModelTester:


@require_torch
+@slow
 class MaskFormerModelTest(ModelTesterMixin, unittest.TestCase):

    all_model_classes = (MaskFormerModel, MaskFormerForInstanceSegmentation) if is_torch_available() else ()