Thank you for this example!I had no problem when running in databricks, but for anyone trying to run it locally, you'll have to instal the pyspark-data-sources
see https://github.com/allisonwang-db/pyspark-data-sources/blob/master/README.md
