Fix replica identity mismatch for partitioned tables with publish_via_partition_root

Fix replica identity mismatch for partitioned tables with publish_via_partition_root

Edit