test-prefect-logging.py

from prefect import Flow, task
from loguru import logger


def fn1():
    print("Hello from fn1")


def fn2():
    logger.info("Hello from fn2")


@task(log_stdout=True)
def task():
    print("Hello from task")
    fn1()
    fn2()


with Flow("test") as flow:
    _ = task()