How to convert MaterializedDatasets to Datasets

Hi, we got two MaterializedDatasets after calling data.train_test_split, but most ml trainers accepts Datasets as arguments, so how to do the conversation?