354ded3b6596154814f6e8cd3352c9f4e8358344,src/sagemaker/amazon/kmeans.py,KMeansModel,__init__,#KMeansModel#Any#Any#Any#,133

Before Change

    def __init__(self, model_data, role, sagemaker_session=None):
        sagemaker_session = sagemaker_session or Session()
        image = registry(sagemaker_session.boto_session.region_name) + "/" + KMeans.repo
        super(KMeansModel, self).__init__(model_data, image, role, predictor_cls=KMeansPredictor,
                                          sagemaker_session=sagemaker_session)

After Change

    def __init__(self, model_data, role, sagemaker_session=None):
        sagemaker_session = sagemaker_session or Session()
        repo = "{}:{}".format(KMeans.repo_name, KMeans.repo_version)
        image = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
        super(KMeansModel, self).__init__(model_data, image, role, predictor_cls=KMeansPredictor,
                                          sagemaker_session=sagemaker_session)

Instances