The human-centric platform for production ML & AI

Outerbounds

Hi folks, does metaflow allow configuring node affinity for tasks running on kubernetes? We have flows with steps that should run on a specific node group: I noticed that the `@kubernetes` decorator accepts tolerations but I can't see anything about node affinity... Is this supported? How do people go about doing this today?

:pika-detective:  There appears to be a `node_selector` in the <https://github.com/Netflix/metaflow/blame/f165ea366f791222525d1d61b5f100bfe4865839/metaflow/plugins/kubernetes/kubernetes_decorator.py#L100C21-L100C21|decorator definition> that is <https://docs.metaflow.org/api/step-decorators/kubernetes|not in the docs>?...

Nice detective work <@U05NYD9T16J>
I was also looking for node affinity and only found tolerations in the doc.

So does it mean that configuring node affinity is supported? If so, will the docs be updated at some point?