Add default values and shapes for few partitions