def build(run, project, model_type):
    docker = Docker()
    docker_path = os.path.join(ROOT_DIR, 'sagemaker', model_type)
    image_name = get_image_name(run, project)
    docker.build(docker_path, image_name)
def build(run, project, model_type):
    docker = Docker()
    docker_path = os.path.join(ROOT_DIR, 'sagemaker', model_type)
    image_name = get_image_name(run, project)
    docker.build(docker_path, image_name)