3857327e011ccf7095613d5d01843cc36e67ac37,tests/integ/test_mxnet_train.py,,test_deploy_model_with_update_endpoint,#Any#Any#Any#,155

Before Change

        assert old_config_name != new_config_name
        assert new_production_variants["InstanceType"] == "ml.m4.xlarge"
        assert new_production_variants["InitialInstanceCount"] == 1
        assert new_production_variants["AcceleratorType"] is None


@pytest.mark.skip(
    reason="This test has always failed, but the failure was masked by a bug. "

After Change

        // Wait for endpoint to finish updating
        max_retry_count = 40  // Endpoint update takes ~7min. 40 retries * 30s sleeps = 20min timeout
        current_retry_count = 0
        while current_retry_count <= max_retry_count:
            if current_retry_count >= max_retry_count:
                raise Exception("Endpoint status not "InService" within expected timeout.")
            time.sleep(30)
            new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
                EndpointName=endpoint_name
            )
            current_retry_count += 1
            if new_endpoint["EndpointStatus"] == "InService":
                break

        new_config_name = new_endpoint["EndpointConfigName"]
        new_config = sagemaker_session.sagemaker_client.describe_endpoint_config(
            EndpointConfigName=new_config_name
        )

Instances