import CachedDataset in __init__