example.nomad

job "toaster-mnist" {
    datacenters = ["us-aep-1-prod-multipaas-1"]
    type = "batch"

    group "toaster-mnist-group" {

        task "toaster-mnist-task" {

            driver = "docker"

            config {
                image   = "dl2.homeawaycorp.com/rdickerson/toaster-mnist:0.0.6"
                command = "/bin/bash"
                args    = ["-c", "make train"]
            }

            resources {
                cpu     = 16000
                memory  = 8000
            }

            env {
                MLFLOW_TRACKING_URI = "http://10.90.112.173:27309"
                MLFLOW_TRACKING_USERNAME = "rdickerson"
                MLFLOW_EXPERIMENT_NAME = "toaster-mnist"
                learning_rate = "0.1"
            }

            vault {
                change_mode     = "signal"
                change_signal   = "SIGHUP"
                env             = "true"
                policies        = ["datascience"]
            }

            template {
                data = <<EOH
                {{ with secret "datascience/mlflow" }}
                AWS_ACCESS_KEY_ID={{ .Data.AccessKeyId }}
                AWS_SECRET_ACCESS_KEY={{ .Data.SecretAccessKey }}
                {{ end }}
                EOH
                destination = "secrets/secrets.txt",
                env = true
            }
        } 
    } 
}