๐Ÿ“ฆ anna-geller / packaging-prefect-flows

๐Ÿ“„ github_kubernetes_run.py ยท 25 lines
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25from prefect import Flow, task
from prefect.storage import GitHub
from prefect.run_configs import KubernetesRun


FLOW_NAME = "github_kubernetes_run"
STORAGE = GitHub(
    repo="anna-geller/packaging-prefect-flows",
    path=f"flows/{FLOW_NAME}.py",
    access_token_secret="GITHUB_ACCESS_TOKEN",  # required with private repositories
)


@task(log_stdout=True)
def hello_world():
    text = f"hello from {FLOW_NAME}"
    print(text)
    return text


with Flow(
    FLOW_NAME, storage=STORAGE, run_config=KubernetesRun(labels=["k8s"],),
) as flow:
    hw = hello_world()