Add "partitioned by" to ensemble model
Apply partitioned by annotation on the ensemble class