change: add tests to quick canary (#1228)

Author: knakad

tests/integ/test_auto_ml.py

@@ -49,6 +49,7 @@
     tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS,
     reason="AutoML is not supported in the region yet.",
 )
+@pytest.mark.canary_quick
 def test_auto_ml_fit(sagemaker_session):
     auto_ml = AutoML(
         role=ROLE,
@@ -215,6 +216,7 @@ def test_best_candidate(sagemaker_session):
     tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS,
     reason="AutoML is not supported in the region yet.",
 )
+@pytest.mark.canary_quick
 def test_deploy_best_candidate(sagemaker_session, cpu_instance_type):
     auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
 

tests/integ/test_debugger.py

@@ -15,6 +15,8 @@
 import os
 import uuid
 
+import pytest
+
 from sagemaker.debugger import Rule
 from sagemaker.debugger import DebuggerHookConfig
 from sagemaker.debugger import TensorBoardOutputConfig
@@ -365,6 +367,7 @@ def test_mxnet_with_tensorboard_output_config(
         _wait_and_assert_that_no_rule_jobs_errored(training_job=mx.latest_training_job)
 
 
+@pytest.mark.canary_quick
 def test_mxnet_with_all_rules_and_configs(sagemaker_session, mxnet_full_version, cpu_instance_type):
     with timeout(minutes=TRAINING_DEFAULT_TIMEOUT_MINUTES):
         rules = [

tests/integ/test_experiments_analytics.py

@@ -3,9 +3,12 @@
 import uuid
 import time
 
+import pytest
+
 from sagemaker.analytics import ExperimentAnalytics
 
 
+@pytest.mark.canary_quick
 def test_experiment_analytics(sagemaker_session):
     sm = sagemaker_session.sagemaker_client
 

tests/integ/test_model_monitor.py

@@ -14,9 +14,9 @@
 
 import json
 import os
+import uuid
 
 import pytest
-import uuid
 
 import tests.integ
 import tests.integ.timeout
@@ -270,6 +270,7 @@ def updated_output_kms_key(sagemaker_session):
     tests.integ.test_region() in tests.integ.NO_MODEL_MONITORING_REGIONS,
     reason="ModelMonitoring is not yet supported in this region.",
 )
+@pytest.mark.canary_quick
 def test_default_monitor_suggest_baseline_and_create_monitoring_schedule_with_customizations(
     sagemaker_session, output_kms_key, volume_kms_key, predictor
 ):

tests/integ/test_processing.py

@@ -133,6 +133,7 @@ def test_sklearn(sagemaker_session, sklearn_full_version, cpu_instance_type):
     assert ROLE in job_description["RoleArn"]
 
 
+@pytest.mark.canary_quick
 def test_sklearn_with_customizations(
     sagemaker_session, image_uri, sklearn_full_version, cpu_instance_type, output_kms_key
 ):
@@ -347,6 +348,7 @@ def test_sklearn_with_no_inputs_or_outputs(
     assert job_description["StoppingCondition"] == {"MaxRuntimeInSeconds": 3600}
 
 
+@pytest.mark.canary_quick
 def test_script_processor(sagemaker_session, image_uri, cpu_instance_type, output_kms_key):
     input_file_path = os.path.join(DATA_DIR, "dummy_input.txt")
 
@@ -474,6 +476,7 @@ def test_script_processor_with_no_inputs_or_outputs(
     assert job_description["StoppingCondition"] == {"MaxRuntimeInSeconds": 3600}
 
 
+@pytest.mark.canary_quick
 def test_processor(sagemaker_session, image_uri, cpu_instance_type, output_kms_key):
     script_path = os.path.join(DATA_DIR, "dummy_script.py")