add extra partition states for PartitionedSignal assign