Skip to content

Commit

Permalink
Deprecate SSD Resnet Tests (#1739)
Browse files Browse the repository at this point in the history
* skip ssd resnet test

* Update test_ssd_resnet.py

* quality
  • Loading branch information
Satrat committed Sep 28, 2023
1 parent adb0eca commit 1174f8a
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions tests/sparseml/pytorch/models/detection/test_ssd_resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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",
[
Expand Down

0 comments on commit 1174f8a

Please sign in to comment.