def _model_dir_with_training_job()

in src/sagemaker_tensorflow_container/training.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def _model_dir_with_training_job(model_dir, job_name):
    if model_dir and model_dir.startswith("/opt/ml"):
        return model_dir
    else:
        return "{}/{}/model".format(model_dir, job_name)