Bring your towel and join one of the fastest growing data communities. Welcome to our second-generation open source orchestration platform, a completely rethought approach to dataflow automation.

Prefect Community

Hi there, I am trying to implement a small proof of concept project using Prefect

Hi <@USYSZBXDY>! What problem are you having?

hi <@UN6FTLFAS> I've detailed the problem below! Thanks for your interest

Of course! Can you try passing `train` directly to the `train_model` task and looking for the result there by accessing `train.category_id` within the task?

yes, thats exactly what i did to get around!

however this breaks a bit the abstraction, as this function should receive only the labels, not all the data in train

If I would like to do that then I should probably create a new task such as `labels = extract_labels(train)`. Right?

```def extract_labels (train):
    return train.category_id```

Yes that would work nicely, then you can pass that directly to `train_model`