Allow shorthand partition shuffle to specify only key class.