def _initialize(self, boto_session, sagemaker_client,
sagemaker_runtime_client, **kwargs): # pylint: disable=unused-argument
"""Initialize this Local SageMaker Session.
Args:
boto_session:
sagemaker_client:
sagemaker_runtime_client:
kwargs:
Returns:
"""
if boto_session is None:
self.boto_session = boto3.Session()
else:
self.boto_session = boto_session
# self.boto_session = boto_session or boto3.Session()
self._region_name = self.boto_session.region_name
if self._region_name is None:
raise ValueError(
"Must setup local AWS configuration with a region supported by SageMaker."
)
self.sagemaker_client = LocalSagemakerClient(self)
self.sagemaker_runtime_client = LocalSagemakerRuntimeClient(
self.config)
self.local_mode = True
if self.s3_endpoint_url is not None:
self.s3_resource = boto_session.resource(
"s3", endpoint_url=self.s3_endpoint_url)
self.s3_client = boto_session.client(
"s3", endpoint_url=self.s3_endpoint_url)
sagemaker_config_file = os.path.join(os.path.expanduser("~"),
".sagemaker", "config.yaml")
if os.path.exists(sagemaker_config_file):
try:
import yaml
except ImportError as e:
logger.error(
_module_import_error("yaml", "Local mode", "local"))
raise e
self.config = yaml.load(open(sagemaker_config_file, "r"))
if self._disable_local_code and "local" in self.config:
self.config["local"]["local_code"] = False
def __init__(self, config=None):
"""Initializes a LocalSageMakerRuntimeClient
Args:
config (dict): Optional configuration for this client. In particular only
the local port is read.
"""
try:
import urllib3
except ImportError as e:
logging.error(_module_import_error("urllib3", "Local mode", "local"))
raise e
self.http = urllib3.PoolManager()
self.serving_port = 8080
self.config = config
self.serving_port = get_config_value("local.serving_port", config) or 8080
def _initialize(self, boto_session, sagemaker_client,
sagemaker_runtime_client):
"""Initialize this Local SageMaker Session.
Args:
boto_session:
sagemaker_client:
sagemaker_runtime_client:
Returns:
"""
self.boto_session = boto_session or boto3.Session()
self._region_name = self.boto_session.region_name
if self._region_name is None:
raise ValueError(
"Must setup local AWS configuration with a region supported by SageMaker."
)
self.sagemaker_client = LocalSagemakerClient(self)
self.sagemaker_runtime_client = LocalSagemakerRuntimeClient(
self.config)
self.local_mode = True
sagemaker_config_file = os.path.join(os.path.expanduser("~"),
".sagemaker", "config.yaml")
if os.path.exists(sagemaker_config_file):
try:
import yaml
except ImportError as e:
logging.error(
_module_import_error("yaml", "Local mode", "local"))
raise e
self.config = yaml.load(open(sagemaker_config_file, "r"))
