fix integ test and use list comprehension instead of map

Author: laurenyu

src/sagemaker/predictor.py

@@ -303,7 +303,7 @@ def _get_model_names(self):
             EndpointConfigName=self._endpoint_config_name
         )
         production_variants = endpoint_config["ProductionVariants"]
-        return map(lambda d: d["ModelName"], production_variants)
+        return [d["ModelName"] for d in production_variants]
 
 
 class _CsvSerializer(object):

tests/integ/test_mxnet_train.py

@@ -81,9 +81,8 @@ def _deploy_estimator_and_assert_instance_type(estimator, instance_type):
                 predictor = estimator.deploy(1, instance_type)
 
                 model_name = predictor._model_names[0]
-                endpoint_name = predictor.endpoint
                 config_name = sagemaker_session.sagemaker_client.describe_endpoint(
-                    EndpointName=endpoint_name
+                    EndpointName=predictor.endpoint_name
                 )["EndpointConfigName"]
                 config = sagemaker_session.sagemaker_client.describe_endpoint_config(
                     EndpointConfigName=config_name
@@ -94,7 +93,7 @@ def _deploy_estimator_and_assert_instance_type(estimator, instance_type):
 
         assert config["ProductionVariants"][0]["InstanceType"] == instance_type
 
-        return (model_name, endpoint_name, config_name)
+        return (model_name, predictor.endpoint_name, config_name)
 
     estimator = MXNet.attach(mxnet_training_job, sagemaker_session)
     estimator.base_job_name = "test-mxnet-deploy-twice"