fix sanity test failure

Author: Jinpeng Qi

docker/build_artifacts/sagemaker/python_service.py

@@ -69,8 +69,10 @@ def __init__(self):
             # during the _handle_load_model_post()
             self.model_handlers = {}
         else:
-            self._tfs_grpc_ports = self._reconstruct_sagemaker_selected_ports(TFS_GRPC_SELECTED_PORTS)
-            self._tfs_rest_ports = self._reconstruct_sagemaker_selected_ports(TFS_REST_SELECTED_PORTS)
+            self._tfs_grpc_ports = self._reconstruct_sagemaker_selected_ports(
+                TFS_GRPC_SELECTED_PORTS)
+            self._tfs_rest_ports = self._reconstruct_sagemaker_selected_ports(
+                TFS_REST_SELECTED_PORTS)
 
             self._channels = {}
             for grpc_port in self._tfs_grpc_ports:

docker/build_artifacts/sagemaker/serve.py

@@ -97,9 +97,9 @@ def __init__(self):
             # select non-overlapping grpc and rest ports based on tfs instance count
             # record the init and end value in order to reconstruct them in python service
             self._tfs_grpc_selected_ports = "{}-{}".format(low,
-                                                       low + 2 * self._tfs_instance_count)
+                                                           low + 2 * self._tfs_instance_count)
             self._tfs_rest_selected_ports = "{}-{}".format(low + 1,
-                                                       low + 2 * self._tfs_instance_count + 1)
+                                                           low + 2 * self._tfs_instance_count + 1)
             for i in range(self._tfs_instance_count):
                 self._tfs_grpc_port.append(str(low + 2 * i))
                 self._tfs_rest_port.append(str(low + 2 * i + 1))
@@ -199,7 +199,8 @@ def _setup_gunicorn(self):
             "python_service:app").format(self._gunicorn_worker_class,
                                          self._gunicorn_workers, self._gunicorn_threads,
                                          python_path_option, ",".join(python_path_content),
-                                         self._tfs_grpc_selected_ports, self._tfs_rest_selected_ports,
+                                         self._tfs_grpc_selected_ports,
+                                         self._tfs_rest_selected_ports,
                                          self._tfs_enable_multi_model_endpoint,
                                          self._sagemaker_port_range,
                                          self._tfs_wait_time_seconds)