loguruflow.py

from prefect import flow, task
from loguru import logger
from prefect_utils.loguru import enable_loguru_support


def mycustom_function():
    logger.info("I'm a custom function")


@task
def mytask():
    enable_loguru_support()
    logger.info("This task log shows up in the Prefect UI")


@flow
def myflow():
    enable_loguru_support()
    logger.info("This flow log shows up in the Prefect UI")
    mytask()
    mycustom_function()
    mytask.submit()


if __name__ == "__main__":
    myflow()