diff --git a/Makefile b/Makefile index d0ed1f899dc..fb210d42003 100644 --- a/Makefile +++ b/Makefile @@ -123,7 +123,7 @@ install-models-hub-torch: test-torch: - pytest ${COVERAGE_ARGS} tests/torch -m "not weekly and not nightly and not models_hub_test" --junitxml ${JUNITXML_PATH} $(DATA_ARG) + pytest ${COVERAGE_ARGS} tests/torch -m "not weekly and not nightly and not models_hub" --junitxml ${JUNITXML_PATH} $(DATA_ARG) test-torch-nightly: pytest ${COVERAGE_ARGS} tests/torch -m nightly --junitxml ${JUNITXML_PATH} $(DATA_ARG) diff --git a/tests/torch/models_hub_test/common.py b/tests/torch/models_hub_test/common.py index 3f607f79272..c7744d77761 100644 --- a/tests/torch/models_hub_test/common.py +++ b/tests/torch/models_hub_test/common.py @@ -24,6 +24,7 @@ from nncf.torch.model_creation import wrap_model +@pytest.mark.models_hub class BaseTestModel(ABC): @abstractmethod def load_model(self, model_name: str): diff --git a/tests/torch/models_hub_test/test_timm.py b/tests/torch/models_hub_test/test_timm.py index 38468bffd7c..f616407cfff 100644 --- a/tests/torch/models_hub_test/test_timm.py +++ b/tests/torch/models_hub_test/test_timm.py @@ -42,7 +42,6 @@ def get_all_models() -> list: return filter_timm(m_list) -@pytest.mark.models_hub class TestTimmModel(BaseTestModel): def load_model(self, model_name: str): m = timm.create_model(model_name, pretrained=False)