Skip to content

Commit 5b37502

Browse files
laurenyulaurenyu
committed
breaking: rename image to image_uri for models
1 parent 5dee4e4 commit 5b37502

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

63 files changed

+310
-319
lines changed

‎src/sagemaker/amazon/amazon_estimator.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ class constructor
156156
init_params[attribute] = init_params["hyperparameters"][value.name]
157157

158158
del init_params["hyperparameters"]
159-
del init_params["image"]
159+
del init_params["image_uri"]
160160
return init_params
161161

162162
def prepare_workflow_for_training(self, records=None, mini_batch_size=None, job_name=None):

‎src/sagemaker/amazon/factorization_machines.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,9 +312,9 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
312312
"""
313313
sagemaker_session = sagemaker_session or Session()
314314
repo = "{}:{}".format(FactorizationMachines.repo_name, FactorizationMachines.repo_version)
315-
image = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
315+
image_uri = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
316316
super(FactorizationMachinesModel, self).__init__(
317-
image,
317+
image_uri,
318318
model_data,
319319
role,
320320
predictor_cls=FactorizationMachinesPredictor,

‎src/sagemaker/amazon/ipinsights.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,12 +218,12 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
218218
"""
219219
sagemaker_session = sagemaker_session or Session()
220220
repo = "{}:{}".format(IPInsights.repo_name, IPInsights.repo_version)
221-
image = "{}/{}".format(
221+
image_uri = "{}/{}".format(
222222
registry(sagemaker_session.boto_session.region_name, IPInsights.repo_name), repo
223223
)
224224

225225
super(IPInsightsModel, self).__init__(
226-
image,
226+
image_uri,
227227
model_data,
228228
role,
229229
predictor_cls=IPInsightsPredictor,

‎src/sagemaker/amazon/kmeans.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,9 +243,9 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
243243
"""
244244
sagemaker_session = sagemaker_session or Session()
245245
repo = "{}:{}".format(KMeans.repo_name, KMeans.repo_version)
246-
image = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
246+
image_uri = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
247247
super(KMeansModel, self).__init__(
248-
image,
248+
image_uri,
249249
model_data,
250250
role,
251251
predictor_cls=KMeansPredictor,

‎src/sagemaker/amazon/lda.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -215,11 +215,11 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
215215
"""
216216
sagemaker_session = sagemaker_session or Session()
217217
repo = "{}:{}".format(LDA.repo_name, LDA.repo_version)
218-
image = "{}/{}".format(
218+
image_uri = "{}/{}".format(
219219
registry(sagemaker_session.boto_session.region_name, LDA.repo_name), repo
220220
)
221221
super(LDAModel, self).__init__(
222-
image,
222+
image_uri,
223223
model_data,
224224
role,
225225
predictor_cls=LDAPredictor,

‎src/sagemaker/amazon/linear_learner.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -476,9 +476,9 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
476476
"""
477477
sagemaker_session = sagemaker_session or Session()
478478
repo = "{}:{}".format(LinearLearner.repo_name, LinearLearner.repo_version)
479-
image = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
479+
image_uri = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
480480
super(LinearLearnerModel, self).__init__(
481-
image,
481+
image_uri,
482482
model_data,
483483
role,
484484
predictor_cls=LinearLearnerPredictor,

‎src/sagemaker/amazon/ntm.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,11 +245,11 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
245245
"""
246246
sagemaker_session = sagemaker_session or Session()
247247
repo = "{}:{}".format(NTM.repo_name, NTM.repo_version)
248-
image = "{}/{}".format(
248+
image_uri = "{}/{}".format(
249249
registry(sagemaker_session.boto_session.region_name, NTM.repo_name), repo
250250
)
251251
super(NTMModel, self).__init__(
252-
image,
252+
image_uri,
253253
model_data,
254254
role,
255255
predictor_cls=NTMPredictor,

‎src/sagemaker/amazon/object2vec.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -351,11 +351,11 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
351351
"""
352352
sagemaker_session = sagemaker_session or Session()
353353
repo = "{}:{}".format(Object2Vec.repo_name, Object2Vec.repo_version)
354-
image = "{}/{}".format(
354+
image_uri = "{}/{}".format(
355355
registry(sagemaker_session.boto_session.region_name, Object2Vec.repo_name), repo
356356
)
357357
super(Object2VecModel, self).__init__(
358-
image,
358+
image_uri,
359359
model_data,
360360
role,
361361
predictor_cls=Predictor,

‎src/sagemaker/amazon/pca.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,9 +227,9 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
227227
"""
228228
sagemaker_session = sagemaker_session or Session()
229229
repo = "{}:{}".format(PCA.repo_name, PCA.repo_version)
230-
image = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
230+
image_uri = "{}/{}".format(registry(sagemaker_session.boto_session.region_name), repo)
231231
super(PCAModel, self).__init__(
232-
image,
232+
image_uri,
233233
model_data,
234234
role,
235235
predictor_cls=PCAPredictor,

‎src/sagemaker/amazon/randomcutforest.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,11 +206,11 @@ def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
206206
"""
207207
sagemaker_session = sagemaker_session or Session()
208208
repo = "{}:{}".format(RandomCutForest.repo_name, RandomCutForest.repo_version)
209-
image = "{}/{}".format(
209+
image_uri = "{}/{}".format(
210210
registry(sagemaker_session.boto_session.region_name, RandomCutForest.repo_name), repo
211211
)
212212
super(RandomCutForestModel, self).__init__(
213-
image,
213+
image_uri,
214214
model_data,
215215
role,
216216
predictor_cls=RandomCutForestPredictor,

0 commit comments

Comments
 (0)