Merge branch 'master' of github.com:ChuyangDeng/sagemaker-tensorflow-serving-container

Author: Chuyang Deng

docker/build_artifacts/sagemaker/python_service.py

@@ -222,9 +222,7 @@ def _import_custom_modules(self, model_name):
 
         if os.path.exists(inference_script_path):
             handler, input_handler, output_handler = self._import_handlers(model_name)
-            model_handlers = self._make_handler(handler,
-                                                input_handler,
-                                                output_handler)
+            model_handlers = self._make_handler(handler, input_handler, output_handler)
             self.model_handlers[model_name] = model_handlers
         else:
             self.model_handlers[model_name] = default_handler

test/integration/local/test_multi_model_endpoint.py

@@ -20,8 +20,12 @@
 import pytest
 import requests
 
-from multi_model_endpoint_test_utils import make_invocation_request, make_list_model_request,\
-    make_load_model_request, make_unload_model_request
+from multi_model_endpoint_test_utils import (
+    make_invocation_request,
+    make_list_model_request,
+    make_load_model_request,
+    make_unload_model_request,
+)
 
 PING_URL = 'http://localhost:8080/ping'