from prefect import task, Flow
import FilterTask
from prefect.backend import FlowRunView

import prefect

@task
def get_list():
    return [1, 2, 3, 4, 5, 6]


@task
def print_number(x):
    print(x)


@task()
def get_list_results():
    logger = prefect.context.get("logger")
    flow_run_id = prefect.context.get("flow_run_id")
    flow_run = FlowRunView.from_flow_run_id(flow_run_id)

    get_list_task = flow_run.get_task_run("get_list-1")
    <http://logger.info|logger.info>(get_list_task.get_result())


with Flow("test") as flow:
    numbers = get_list()
    prints = print_number.map(numbers)
    get_list_results(upstream_tasks=[prints])