def _format_model_uri_input(model_uri, validate_uri=True):
"""
Args:
model_uri:
validate_uri:
"""
if isinstance(model_uri, string_types
) and validate_uri and model_uri.startswith("s3://"):
return s3_input(
model_uri,
input_mode="File",
distribution="FullyReplicated",
content_type="application/x-sagemaker-model",
)
if isinstance(model_uri, string_types
) and validate_uri and model_uri.startswith("file://"):
return file_input(model_uri)
if isinstance(model_uri, string_types) and validate_uri:
raise ValueError(
'Model URI must be a valid S3 or FILE URI: must start with "s3://" or '
'"file://')
if isinstance(model_uri, string_types):
return s3_input(
model_uri,
input_mode="File",
distribution="FullyReplicated",
content_type="application/x-sagemaker-model",
)
raise ValueError(
"Cannot format model URI {}. Expecting str".format(model_uri))
def _format_string_uri_input(uri_input, validate_uri=True, content_type=None, input_mode=None):
"""
Args:
uri_input:
validate_uri:
content_type:
input_mode:
"""
if isinstance(uri_input, str) and validate_uri and uri_input.startswith("s3://"):
return s3_input(uri_input, content_type=content_type, input_mode=input_mode)
if isinstance(uri_input, str) and validate_uri and uri_input.startswith("file://"):
return file_input(uri_input)
if isinstance(uri_input, str) and validate_uri:
raise ValueError(
'URI input {} must be a valid S3 or FILE URI: must start with "s3://" or '
'"file://"'.format(uri_input)
)
if isinstance(uri_input, str):
return s3_input(uri_input, content_type=content_type, input_mode=input_mode)
if isinstance(uri_input, s3_input):
return uri_input
if isinstance(uri_input, file_input):
return uri_input
raise ValueError(
"Cannot format input {}. Expecting one of str, s3_input, or file_input".format(
uri_input
)
)
def _format_string_uri_input(
uri_input,
validate_uri=True,
content_type=None,
input_mode=None,
compression=None,
target_attribute_name=None,
):
"""
Args:
uri_input:
validate_uri:
content_type:
input_mode:
compression:
target_attribute_name:
"""
if isinstance(uri_input,
str) and validate_uri and uri_input.startswith("s3://"):
s3_input_result = s3_input(
uri_input,
content_type=content_type,
input_mode=input_mode,
compression=compression,
target_attribute_name=target_attribute_name,
)
return s3_input_result
if isinstance(
uri_input,
str) and validate_uri and uri_input.startswith("file://"):
return file_input(uri_input)
if isinstance(uri_input, str) and validate_uri:
raise ValueError(
'URI input {} must be a valid S3 or FILE URI: must start with "s3://" or '
'"file://"'.format(uri_input))
if isinstance(uri_input, str):
s3_input_result = s3_input(
uri_input,
content_type=content_type,
input_mode=input_mode,
compression=compression,
target_attribute_name=target_attribute_name,
)
return s3_input_result
if isinstance(uri_input, (s3_input, file_input, FileSystemInput)):
return uri_input
raise ValueError(
"Cannot format input {}. Expecting one of str, s3_input, file_input or "
"FileSystemInput".format(uri_input))
def _format_string_uri_input(input):
if isinstance(input, str):
if input.startswith('s3://'):
return s3_input(input)
elif input.startswith('file://'):
return file_input(input)
else:
raise ValueError('Training input data must be a valid S3 or FILE URI: must start with "s3://" or '
'"file://"')
elif isinstance(input, s3_input):
return input
elif isinstance(input, file_input):
return input
else:
raise ValueError('Cannot format input {}. Expecting one of str, s3_input, or file_input'.format(input))
def _format_model_uri_input(model_uri):
if isinstance(model_uri, string_types):
if model_uri.startswith('s3://'):
return s3_input(model_uri,
input_mode='File',
distribution='FullyReplicated',
content_type='application/x-sagemaker-model')
elif model_uri.startswith('file://'):
return file_input(model_uri)
else:
raise ValueError(
'Model URI must be a valid S3 or FILE URI: must start with "s3://" or '
'"file://')
else:
raise ValueError(
'Cannot format model URI {}. Expecting str'.format(model_uri))
