k8s_container = {
    "name": "prefect-job",
    "env": [
        {
            "name": "SECRET_API_KEY",
            "valueFrom": {
                "secretKeyRef": {
                    "name": "secret-auth",
                    "key": "password",
                }
            },
        }
    ]
}

KubernetesJob(
    namespace="default",
    image=f"{os.environ['AWS_ECR_URL']}/prefect",
    image_pull_policy=KubernetesImagePullPolicy.ALWAYS,
    # Base job template:
    # <https://github.com/PrefectHQ/prefect/blob/main/src/prefect/infrastructure/kubernetes.py#L142>
    job={
        "apiVersion": "batch/v1",
        "kind": "Job",
        "metadata": {"labels": {}},
        "spec": {
            "ttlSecondsAfterFinished": 1800,
            "template": {
                "spec": {
                    "restartPolicy": "Never",
                    "parallelism": 1,
                    "completions": 1,
                    "containers": [k8s_container],
                    "affinity": {
                        "nodeAffinity": {
                            "preferredDuringSchedulingIgnoredDuringExecution": [
                                {
                                    "weight": 1,
                                    "preference": {
                                        "matchExpressions": [
                                            {
                                                "key": "nodegroup-name",
                                                "operator": "In",
                                                "values": ["compute"],
                                            }
                                        ]
                                    },
                                }
                            ]
                        }
                    },
                }
            },
        },
    },
    stream_output=True,
    pod_watch_timeout_seconds=60,
    job_watch_timeout_seconds=60,
)

KubernetesJob

command