https://platoaistream.net/plato-data/create-random-and-stratified-samples-of-data-with-amazon-sagemaker-data-wrangler/