def test_log_model_missing_warning_wrong_parent_dir(logger):
training._log_model_missing_warning(
os.path.join(RESOURCE_PATH, 'test_dir_wrong_parent_dir'))
logger.warn.assert_called_with(
'Your model will NOT be servable with SageMaker TensorFlow Serving containers.'
'The SavedModel bundle is under directory \"{}\", not a numeric name.'.
format('not-digit'))
def test_log_model_missing_warning_wrong_format(logger):
training._log_model_missing_warning(
os.path.join(RESOURCE_PATH, 'test_dir_wrong_model'))
logger.warn.assert_called_with(
'Your model will NOT be servable with SageMaker TensorFlow Serving container.'
'The model artifact was not saved in the TensorFlow '
'SavedModel directory structure:\n'
'https://www.tensorflow.org/guide/saved_model#structure_of_a_savedmodel_directory'
)
def test_log_model_missing_warning_no_model(logger):
path = os.path.join(RESOURCE_PATH, 'test_dir_empty')
if not os.path.exists(path):
os.mkdir(path)
training._log_model_missing_warning(path)
logger.warn.assert_called_with('No model artifact is saved under path {}.'
' Your training job will not save any model files to S3.\n'
'For details of how to construct your training script see:\n'
'https://sagemaker.readthedocs.io/en/stable/using_tf.html#adapting-your-local-tensorflow-script' # noqa
.format(path))
def test_log_model_missing_warning_no_model(logger):
path = os.path.join(RESOURCE_PATH, 'test_dir_empty')
if not os.path.exists(path):
os.mkdir(path)
training._log_model_missing_warning(path)
logger.warn.assert_called_with(
'No model artifact is saved under path {}.'
' Your training job will not save any model files to S3.\n'
'For details of how to construct your training script see:\n'
'https://github.com/aws/sagemaker-python-sdk/tree/master/src/sagemaker/tensorflow#adapting-your-local-tensorflow-script' # noqa
.format(path))
def test_log_model_missing_warning_correct(logger):
training._log_model_missing_warning(
os.path.join(RESOURCE_PATH, 'test_dir_correct_model'))
logger.warn.assert_not_called()
