Partition: pass key and object to predicate given context

Noticed this scouting about that this behaviour changed in #1593 and got confused somewhere