fix flake8 (#8)

Author: Dan

src/sagemaker_mxnet_serving_container/default_inference_handler.py

@@ -15,9 +15,9 @@
 import os
 
 import mxnet as mx
-from sagemaker_inference import encoder, errors, decoder, content_types, default_inference_handler
+from sagemaker_inference import content_types, decoder, default_inference_handler, encoder, errors
 
-from sagemaker_mxnet_serving_container.utils import read_data_shapes, get_default_context
+from sagemaker_mxnet_serving_container.utils import get_default_context, read_data_shapes
 
 PREFERRED_BATCH_SIZE_PARAM = 'SAGEMAKER_DEFAULT_MODEL_FIRST_DIMENSION_SIZE'
 INFERENCE_ACCELERATOR_PRESENT_ENV = 'SAGEMAKER_INFERENCE_ACCELERATOR_PRESENT'
@@ -39,7 +39,8 @@ def default_model_fn(self, model_dir, preferred_batch_size=1):
 
         Args:
             model_dir (str): The directory where model files are stored
-            preferred_batch_size (int): The preferred batch size of the model's data shape (default: 1)
+            preferred_batch_size (int): preferred batch size of the model's data shape.
+                Defaults to 1.
 
         Returns:
             mxnet.mod.Module: the loaded model.
@@ -162,8 +163,7 @@ def default_input_fn(self, input_data, content_type, model=None):
             padding = mx.ndarray.zeros(shape=padding_shape)
             ndarray = mx.ndarray.concat(ndarray, padding, dim=0)
 
-        model_input = mx.io.NDArrayIter(ndarray, batch_size=model_batch_size,
-                                        last_batch_handle='pad')
+        model_input = mx.io.NDArrayIter(ndarray, batch_size=model_batch_size, last_batch_handle='pad')
 
         if pad_rows:
             # Update the getpad method on the model_input data iterator to return the amount of

src/sagemaker_mxnet_serving_container/handler_service.py

@@ -19,8 +19,8 @@
 from sagemaker_inference.default_handler_service import DefaultHandlerService
 from sagemaker_inference.transformer import Transformer
 
-from sagemaker_mxnet_serving_container.default_inference_handler import \
-    DefaultMXNetInferenceHandler, DefaultGluonBlockInferenceHandler
+from sagemaker_mxnet_serving_container.default_inference_handler import DefaultGluonBlockInferenceHandler, \
+    DefaultMXNetInferenceHandler
 from sagemaker_mxnet_serving_container.mxnet_module_transformer import MXNetModuleTransformer
 
 

src/sagemaker_mxnet_serving_container/mxnet_module_transformer.py

@@ -15,6 +15,7 @@
 import inspect
 
 from sagemaker_inference.transformer import Transformer
+
 from sagemaker_mxnet_serving_container.default_inference_handler import DefaultModuleInferenceHandler
 
 

src/sagemaker_mxnet_serving_container/utils.py

@@ -13,6 +13,7 @@
 from __future__ import absolute_import
 
 import json
+
 import mxnet as mx
 
 

test/unit/test_default_inference_handler.py

@@ -15,14 +15,13 @@
 import json
 import os
 
+from mock import call, Mock, mock_open, patch
 import mxnet as mx
 import pytest
+from sagemaker_inference import content_types, errors
 
-from mock import patch, Mock, mock_open, call
-from sagemaker_inference import errors, content_types
-
-from sagemaker_mxnet_serving_container.default_inference_handler import DefaultMXNetInferenceHandler, \
-    DefaultModuleInferenceHandler, DefaultGluonBlockInferenceHandler
+from sagemaker_mxnet_serving_container.default_inference_handler import DefaultGluonBlockInferenceHandler, \
+    DefaultModuleInferenceHandler, DefaultMXNetInferenceHandler
 
 MODEL_DIR = 'foo/model'
 
@@ -31,8 +30,7 @@
 
 
 def test_default_mxnet_valid_content_types():
-    assert DefaultMXNetInferenceHandler().VALID_CONTENT_TYPES == \
-           (content_types.JSON, content_types.NPY)
+    assert DefaultMXNetInferenceHandler().VALID_CONTENT_TYPES == (content_types.JSON, content_types.NPY)
 
 
 @patch('mxnet.cpu')
@@ -162,7 +160,7 @@ def test_mxnet_default_output_fn_invalid_content_type():
 
 def test_default_module_valid_content_types():
     assert DefaultModuleInferenceHandler().VALID_CONTENT_TYPES == \
-           (content_types.JSON, content_types.CSV, content_types.NPY)
+        (content_types.JSON, content_types.CSV, content_types.NPY)
 
 
 @patch('mxnet.io.NDArrayIter')
@@ -219,8 +217,7 @@ def test_module_default_input_fn_with_npy(decode, mx_ndarray_iter):
 @patch('mxnet.io.NDArrayIter')
 @patch('sagemaker_inference.decoder.decode', return_value=[0])
 @patch.dict(os.environ, {'SAGEMAKER_INFERENCE_ACCELERATOR_PRESENT': 'true'}, clear=True)
-def test_module_default_input_fn_with_accelerator(decode, mx_ndarray_iter, mx_ndarray,
-                                                              mx_eia):
+def test_module_default_input_fn_with_accelerator(decode, mx_ndarray_iter, mx_ndarray, mx_eia):
     ndarray = Mock(shape=(1, (1,)))
     ndarray.as_in_context.return_value = ndarray
     mx_ndarray.return_value = ndarray

test/unit/test_handler_service.py

@@ -12,9 +12,9 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+from mock import Mock, patch
 import mxnet as mx
 import pytest
-from mock import patch, Mock
 from sagemaker_inference.transformer import Transformer
 
 from sagemaker_mxnet_serving_container.handler_service import HandlerService
@@ -60,7 +60,7 @@ def test_user_module_mxnet_module_transformer(import_module, input_fn, predict_f
 
 
 @patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultMXNetInferenceHandler.default_model_fn')
-@patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultGluonBlockInferenceHandler.default_predict_fn')
+@patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultGluonBlockInferenceHandler.default_predict_fn')  # noqa E501
 @patch('importlib.import_module', return_value=object())
 def test_user_module_mxnet_gluon_transformer(import_module, predict_fn, model_fn):
     model_fn.return_value = mx.gluon.block.Block()

test/unit/test_utils.py

@@ -14,7 +14,7 @@
 
 import json
 
-from mock import Mock, patch, mock_open
+from mock import Mock, mock_open, patch
 
 from sagemaker_mxnet_serving_container.utils import get_default_context, read_data_shapes