fixed unit tests for py27 and integ test

Author: pialidas

src/sagemaker/automl/automl.py

@@ -117,8 +117,8 @@ def attach(cls, job_name, sagemaker_session=None):
 
         _auto_ml_job_desc = sagemaker_session.describe_auto_ml_job(job_name)
         automl_job_tags = sagemaker_session.sagemaker_client.list_tags(
-            _auto_ml_job_desc["AutoMLJobArn"]
-        )
+            ResourceArn=_auto_ml_job_desc["AutoMLJobArn"])["Tags"]
+
         amlj = AutoML(
             role=_auto_ml_job_desc["RoleArn"],
             target_attribute_name=_auto_ml_job_desc["InputDataConfig"][0]["TargetAttributeName"],

tests/integ/test_auto_ml.py

@@ -277,7 +277,10 @@ def test_deploy_best_candidate(sagemaker_session, cpu_instance_type):
     sagemaker_session.sagemaker_client.delete_endpoint(EndpointName=endpoint_name)
 
 
-@pytest.mark.canary_quick
+@pytest.mark.skipif(
+    tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS,
+    reason="AutoML is not supported in the region yet.",
+)
 def test_create_model_best_candidate(sagemaker_session, cpu_instance_type):
     auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
 

tests/unit/sagemaker/automl/test_auto_ml.py

@@ -12,6 +12,8 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+import copy
+
 import pytest
 from mock import Mock, patch
 from sagemaker import AutoML, AutoMLJob, AutoMLInput, CandidateEstimator, PipelineModel
@@ -600,7 +602,7 @@ def test_candidate_estimator_fit(sagemaker_session):
 
 
 def test_validate_and_update_inference_response():
-    cic = CLASSIFICATION_INFERENCE_CONTAINERS.copy()
+    cic = copy.copy(CLASSIFICATION_INFERENCE_CONTAINERS)
 
     AutoML.validate_and_update_inference_response(
         inference_containers=cic,
@@ -622,7 +624,7 @@ def test_validate_and_update_inference_response():
 
 
 def test_validate_and_update_inference_response_wrong_input():
-    cic = CLASSIFICATION_INFERENCE_CONTAINERS.copy()
+    cic = copy.copy(CLASSIFICATION_INFERENCE_CONTAINERS)
 
     with pytest.raises(
         ValueError,
@@ -661,4 +663,4 @@ def test_attach(sagemaker_session):
     assert aml.target_attribute_name == "y"
     assert aml.problem_type == "Auto"
     assert aml.output_path == "s3://output_prefix"
-    assert aml.tags == LIST_TAGS_RESULT
+    assert aml.tags == LIST_TAGS_RESULT["Tags"]