from dask_flow import flow

from prefect.environments.storage import Docker
from prefect.environments import DaskKubernetesEnvironment

flow.storage = Docker(registry_url="gcr.io/execution-layer-demo/flows")
flow.environment = DaskKubernetesEnvironment()
flow.register('execution-layer-demo')