Remove treeAggregatePartitionsWithIndex