Has anyone by chance created a custom dataset to wrap around the Azure ML datasets (https://docs.microsoft.com/en-us/python/api/azureml-core/azureml.core.dataset(class)?view=azure-ml-py) ?