Skip to content

Commit ec53250

Browse files
author
Ignacio Quintero
committed
Fix flake8 and remove unused **kwargs
1 parent 79bac68 commit ec53250

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

src/sagemaker/estimator.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -153,13 +153,12 @@ def fit(self, inputs, wait=True, logs=True, job_name=None):
153153
if wait:
154154
self.latest_training_job.wait(logs=logs)
155155

156-
157156
@classmethod
158157
def _from_training_job(cls, init_params, hyperparameters, image, sagemaker_session):
159158
raise NotImplementedError()
160159

161160
@classmethod
162-
def attach(cls, training_job_name, sagemaker_session=None, **kwargs):
161+
def attach(cls, training_job_name, sagemaker_session=None):
163162
"""Attach to an existing training job.
164163
165164
Create an Estimator bound to an existing training job. After attaching, if

tests/integ/test_linear_learner.py

+1
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def test_linear_learner():
8787
assert record.label["predicted_label"] is not None
8888
assert record.label["score"] is not None
8989

90+
9091
def test_async_linear_learner(sagemaker_session):
9192

9293
training_job_name = ""

tests/integ/test_mxnet_train.py

-1
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,6 @@ def test_async_fit(sagemaker_session):
9090
predictor.predict(data)
9191

9292

93-
9493
def test_deploy_model(mxnet_training_job, sagemaker_session):
9594
endpoint_name = 'test-mxnet-deploy-model-{}'.format(int(time.time()))
9695

0 commit comments

Comments
 (0)