XD-997 Adding initial partitioning support for hdfs-dataset sink