From 1174f8a07d95ab24bbfcbbc55aebbcd848c988c6 Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Thu, 28 Sep 2023 08:54:31 -0400 Subject: [PATCH] Deprecate SSD Resnet Tests (#1739) * skip ssd resnet test * Update test_ssd_resnet.py * quality --- .../pytorch/models/detection/test_ssd_resnet.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/tests/sparseml/pytorch/models/detection/test_ssd_resnet.py b/tests/sparseml/pytorch/models/detection/test_ssd_resnet.py index 352fcf7b6c0..c6399d914ae 100644 --- a/tests/sparseml/pytorch/models/detection/test_ssd_resnet.py +++ b/tests/sparseml/pytorch/models/detection/test_ssd_resnet.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -import os from typing import Callable, Union import pytest @@ -29,14 +28,7 @@ from tests.sparseml.pytorch.models.utils import compare_model -@pytest.mark.skipif( - os.getenv("NM_ML_SKIP_PYTORCH_TESTS", False), - reason="Skipping pytorch tests", -) -@pytest.mark.skipif( - os.getenv("NM_ML_SKIP_MODEL_TESTS", False), - reason="Skipping model tests", -) +@pytest.mark.skip(reason="SSD Resnet models deprecated") @pytest.mark.parametrize( "key,pretrained,pretrained_backbone,test_input,match_const", [