How do I use a different default dataset? Trying to run on Kubeflow where MemoryDataSet won't work. Want to change it to something like PickleDataSet. Seems like DataCatalogWithDefault is no longer in v0.18...