I am trying to build out a tool which uses partitions to put together work items, but for some reason the partition is lost the instant it moves through a switch node (and I'm sure many others). I am using the switch so I can give the users multiple options for defining how the partitioning is done, but losing the partitions at the switch means that I struggle to get anything done correctly down the line.

Did a little playing to see if I could come up with a convoluted scheme to partition the work items, add an attribute to them to store the id for the partition, then expand before the switch so I could repartition them down the line, but even that seems to break it.

How does one work with partitions after the work items are partitioned without them immediately breaking?