Oops, that comment is incorrect. I was using a thrift method where you just specify the values of the partition, not the keys. This should work with a regular dict