Вообще лучше назвать тогда оба варианта `partition_proc` и `partition_keys` как `partition` и смотреть что передали