There are a number of in-progress or needed enhancements to the Kite DatasetSink:
I think it makes sense to tackle these as a more general DatasetSink 2.0 effort.
Update Kite DatasetSink with changes in Kite 0.18.0
Disable the Parquet format in DatasetSink
Add support for Parquet datasets to the DatasetSink
Update Kite dependency to 0.17.0 (or 0.17.1)