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"))