prefect_kubernetes_job_block_flow_deployment.py

1from prefect import flow
2from prefect_kubernetes import KubernetesJob
3
4@flow
5def hello_kubernetes():
6    print("Hello from Kubernetes!")
7
8# Create a KubernetesJob block
9# This block defines the infrastructure settings for the job
10k8s_job = KubernetesJob(
11    image="prefecthq/prefect:2-python3.9",
12    namespace="default",
13    image_pull_policy="IfNotPresent",
14)
15
16if __name__ == "__main__":
17    # Deploy the flow using the KubernetesJob infrastructure
18    # In a real-world scenario, you would use .deploy() or a prefect.yaml file,
19    # but this is the programmatic way to apply infrastructure blocks.
20    hello_kubernetes.deploy(
21        name="k8s-example-deployment",
22        work_pool_name="my-k8s-work-pool", # Assumes you have a work pool of type 'kubernetes'
23        job_variables={"image": "prefecthq/prefect:2-python3.9"}
24    )