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

Hi @Fred Israel! What problem are you having?

hi @nicholas 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!

Awesome! All good now?

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

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

train_model

thanks 🙂