fix removing partition cols from recordbatch logic

fix removing partition cols from recordbatch logic #266