Skip to content

Commit 9e08f50

Browse files
author
Piali Das
committed
Add support for Amazon k-nearest neighbors(KNN) algorithm
1 parent 3d091b4 commit 9e08f50

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/sagemaker/amazon/linear_learner.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -301,11 +301,10 @@ class LinearLearnerModel(Model):
301301
"""Reference LinearLearner s3 model data. Calling :meth:`~sagemaker.model.Model.deploy` creates an Endpoint
302302
and returns a :class:`LinearLearnerPredictor`"""
303303

304-
def __init__(self, model_data, role, sagemaker_session=None, **kwargs):
304+
def __init__(self, model_data, role, sagemaker_session=None):
305305
sagemaker_session = sagemaker_session or Session()
306306
repo = '{}:{}'.format(LinearLearner.repo_name, LinearLearner.repo_version)
307307
image = '{}/{}'.format(registry(sagemaker_session.boto_session.region_name), repo)
308308
super(LinearLearnerModel, self).__init__(model_data, image, role,
309309
predictor_cls=LinearLearnerPredictor,
310-
sagemaker_session=sagemaker_session,
311-
**kwargs)
310+
sagemaker_session=sagemaker_session)

0 commit comments

Comments
 (0)