Fix NoBotoSession

Author: Ignacio Quintero

tests/integ/test_local_mode.py

@@ -18,7 +18,7 @@
 import numpy
 import pytest
 
-from sagemaker.local import LocalSession
+from sagemaker.local import LocalSession, LocalSagemakerRuntimeClient, LocalSagemakerClient
 from sagemaker.mxnet import MXNet
 from sagemaker.tensorflow import TensorFlow
 from tests.integ import DATA_DIR
@@ -35,13 +35,22 @@ class LocalNoS3Session(LocalSession):
     """
     def __init__(self):
         super(LocalSession, self).__init__()
-        self.config = {
-            'local':
-                {
-                    'local_code': True
-                }
-        }
+
+    def _initialize(self, boto_session, sagemaker_client, sagemaker_runtime_client):
+        self.boto_session = None
+        if self.config is None:
+            self.config = {
+                'local':
+                    {
+                        'local_code': True,
+                        'region_name': DEFAULT_REGION
+                    }
+            }
+
         self._region_name = DEFAULT_REGION
+        self.sagemaker_client = LocalSagemakerClient(self)
+        self.sagemaker_runtime_client = LocalSagemakerRuntimeClient(self.config)
+        self.local_mode = True
 
 
 def test_tf_local_mode(tf_full_version, sagemaker_local_session):