Does task decorator currently support python gener...
# ask-community
h
Does task decorator currently support python generators? Trying to see if tasks play nice with yield.
Copy code
@task
def generator(limit: int):
    logger = prefect.context.get("logger")
    for i in range(limit):
        <http://logger.info|logger.info>(i)
        yield i*i
k
We don’t support generators
h
Sounds good, are there plans in Orion to support generators at all?
k
That is a good question. I think so because you can use native Python inside a Flow. I am just not sure if you can use it inside a task
h
Looks like you can use it inside of a task which will be good enough for now!
Copy code
@task
def parse_gen(limit: int):
    logger = prefect.context.get("logger")

    empty = []

    for i in generator(limit):
        if i < 100:
            <http://logger.info|logger.info>(i)
            empty.append(i)

    return empty
👍 1