PySpark custom data source for Hugging Face Datasets - View it on GitHub