test(components): Add shallow_canary marker (#9180)

* Add shallow_canary marker

* Delete trainingV1 from shallow canary

components/aws/sagemaker/tests/integration_tests/component_tests/test_deploy_component.py

@@ -38,7 +38,9 @@ def run_predict_mnist(boto3_session, endpoint_name, download_dir):
     payload = np2csv(train_set[0][30:31])
 
     response = runtime.invoke_endpoint(
-        EndpointName=endpoint_name, ContentType="text/csv", Body=payload,
+        EndpointName=endpoint_name,
+        ContentType="text/csv",
+        Body=payload,
     )
     return json.loads(response["Body"].read().decode())
 
@@ -48,7 +50,8 @@ def run_predict_mnist(boto3_session, endpoint_name, download_dir):
     [
         pytest.param("resources/config/kmeans-mnist-update-endpoint"),
         pytest.param(
-            "resources/config/kmeans-mnist-endpoint", marks=pytest.mark.canary_test
+            "resources/config/kmeans-mnist-endpoint",
+            marks=[pytest.mark.canary_test, pytest.mark.shallow_canary],
         ),
     ],
 )

components/aws/sagemaker/tests/integration_tests/component_tests/test_v2_train_component.py

@@ -9,7 +9,12 @@ import ast
 
 @pytest.mark.parametrize(
     "test_file_dir",
-    [pytest.param("resources/config/ack-training-job", marks=pytest.mark.canary_test)],
+    [
+        pytest.param(
+            "resources/config/ack-training-job",
+            marks=[pytest.mark.canary_test, pytest.mark.shallow_canary],
+        )
+    ],
 )
 def test_trainingjobV2(kfp_client, experiment_id, test_file_dir):
     k8s_client = ack_utils.k8s_client()

components/aws/sagemaker/tests/integration_tests/pytest.ini

@@ -3,4 +3,5 @@ junit_family = xunit2
 addopts = -rA
 markers =
     canary_test: test to be run as part of canaries.
-    fsx_test: tests for FSx features
+    fsx_test: tests for FSx features
+    shallow_canary: a subset of canary_test