fix marks

openvinotoolkit · Nov 16, 2023 · f29ecf9 · f29ecf9
1 parent 4f71bd2
commit f29ecf9
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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)