Added Pytest markers Smoke & Sanity for Runtime tests (#451)

Raghul-M · pre-commit-ci[bot] · web-flow · commit e0cdafece5bf · 2025-07-21T17:23:47.000+05:30
* Added Pytest markers for Runtime tests * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
diff --git a/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_catboost_basic_model_deployment.py b/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_catboost_basic_model_deployment.py
@@ -43,6 +43,7 @@
 )
 
 
+@pytest.mark.sanity
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "mlserver_inference_service", "s3_models_storage_uri", "mlserver_serving_runtime"),
     [
diff --git a/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_lightgbm_basic_model_deployment.py b/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_lightgbm_basic_model_deployment.py
@@ -43,6 +43,7 @@
 )
 
 
+@pytest.mark.smoke
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "mlserver_inference_service", "s3_models_storage_uri", "mlserver_serving_runtime"),
     [
diff --git a/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_mlflow_basic_model_deployment.py b/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_mlflow_basic_model_deployment.py
@@ -43,6 +43,7 @@
 )
 
 
+@pytest.mark.smoke
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "mlserver_inference_service", "s3_models_storage_uri", "mlserver_serving_runtime"),
     [
diff --git a/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_sklearn_basic_model_deployment.py b/tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_sklearn_basic_model_deployment.py
@@ -43,6 +43,7 @@
 )
 
 
+@pytest.mark.sanity
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "mlserver_inference_service", "s3_models_storage_uri", "mlserver_serving_runtime"),
     [
diff --git a/tests/model_serving/model_runtime/triton/basic_model_deployment/test_onnx_model.py b/tests/model_serving/model_runtime/triton/basic_model_deployment/test_onnx_model.py
@@ -32,6 +32,7 @@
 )
 
 
+@pytest.mark.sanity
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "s3_models_storage_uri", "triton_serving_runtime", "triton_inference_service"),
     [
diff --git a/tests/model_serving/model_runtime/triton/basic_model_deployment/test_python_model.py b/tests/model_serving/model_runtime/triton/basic_model_deployment/test_python_model.py
@@ -32,6 +32,7 @@
 )
 
 
+@pytest.mark.smoke
 @pytest.mark.parametrize(
     ("protocol", "model_namespace", "s3_models_storage_uri", "triton_serving_runtime", "triton_inference_service"),
     [

Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,7 @@`
`43`	`43`	`)`
`44`	`44`
`45`	`45`
	`46`	`+@pytest.mark.sanity`
`46`	`47`	`@pytest.mark.parametrize(`
`47`	`48`	`("protocol", "model_namespace", "mlserver_inference_service", "s3_models_storage_uri", "mlserver_serving_runtime"),`
`48`	`49`	`[`
Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,7 @@`
`32`	`32`	`)`
`33`	`33`
`34`	`34`
	`35`	`+@pytest.mark.sanity`
`35`	`36`	`@pytest.mark.parametrize(`
`36`	`37`	`("protocol", "model_namespace", "s3_models_storage_uri", "triton_serving_runtime", "triton_inference_service"),`
`37`	`38`	`[`