diff --git a/tests/model_registry/rest_api/test_model_registry_rest_api.py b/tests/model_registry/rest_api/test_model_registry_rest_api.py index e3053d970..880c4a995 100644 --- a/tests/model_registry/rest_api/test_model_registry_rest_api.py +++ b/tests/model_registry/rest_api/test_model_registry_rest_api.py @@ -30,11 +30,13 @@ {}, {}, MODEL_REGISTER_DATA, + marks=(pytest.mark.smoke), ), pytest.param( {"db_name": "mariadb"}, {"db_name": "mariadb"}, MODEL_REGISTER_DATA, + marks=(pytest.mark.sanity), ), ], indirect=True, @@ -73,7 +75,6 @@ class TestModelRegistryCreationRest: ), ], ) - @pytest.mark.smoke def test_validate_model_registry_resource( self: Self, registered_model_rest_api: dict[str, Any], @@ -86,7 +87,6 @@ def test_validate_model_registry_resource( resource_name=data_key, ) - @pytest.mark.sanity def test_model_registry_validate_api_version( self: Self, model_registry_instance, @@ -100,7 +100,6 @@ def test_model_registry_validate_api_version( expected_version = f"{ModelRegistry.ApiGroup.MODELREGISTRY_OPENDATAHUB_IO}/{ModelRegistry.ApiVersion.V1BETA1}" assert api_version == expected_version - @pytest.mark.sanity def test_model_registry_validate_oauthproxy_enabled( self: Self, model_registry_instance, @@ -143,7 +142,6 @@ def test_model_registry_validate_oauthproxy_enabled( ], indirect=["updated_model_registry_resource"], ) - @pytest.mark.smoke def test_create_update_model_artifact( self, updated_model_registry_resource: dict[str, Any], @@ -189,7 +187,6 @@ def test_create_update_model_artifact( ], indirect=["updated_model_registry_resource"], ) - @pytest.mark.sanity def test_updated_model_version( self, updated_model_registry_resource: dict[str, Any], @@ -235,7 +232,6 @@ def test_updated_model_version( ], indirect=["updated_model_registry_resource"], ) - @pytest.mark.sanity def test_updated_registered_model( self, updated_model_registry_resource: dict[str, Any],