Skip to content

Commit 58eb512

Browse files
author
Balaji Veeramani
committed
Appease lint
1 parent bc48fc7 commit 58eb512

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/unit/test_predictor.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,9 @@ def ret_csv_sagemaker_session():
181181

182182
def test_predict_call_with_csv():
183183
sagemaker_session = ret_csv_sagemaker_session()
184-
predictor = Predictor(ENDPOINT, sagemaker_session, serializer=CSVSerializer(), deserializer=CSVDeserializer())
184+
predictor = Predictor(
185+
ENDPOINT, sagemaker_session, serializer=CSVSerializer(), deserializer=CSVDeserializer()
186+
)
185187

186188
data = [1, 2]
187189
result = predictor.predict(data)
@@ -202,7 +204,9 @@ def test_predict_call_with_csv():
202204

203205
def test_predict_call_with_multiple_accept_types():
204206
sagemaker_session = ret_csv_sagemaker_session()
205-
predictor = Predictor(ENDPOINT, sagemaker_session, serializer=CSVSerializer(), deserializer=StringDeserializer())
207+
predictor = Predictor(
208+
ENDPOINT, sagemaker_session, serializer=CSVSerializer(), deserializer=StringDeserializer()
209+
)
206210

207211
data = [1, 2]
208212
result = predictor.predict(data)

0 commit comments

Comments
 (0)