It's more of a dataset manager at this point and inherits from pytorch dataset without really acting like a dataset.