def METHOD_NAME(train_image, model_artifact_url):
    return {
        "containerHostname": "mnist-kmeans",
        "image": train_image,
        "mode": "SingleModel",
        "modelDataURL": model_artifact_url,
    }