small fix: removing repartition as it has no effect before groupby (#25)

small fix: removing repartition as it has no effect before groupby (#25) #31