def METHOD_NAME(obj, resolver, **_kw):
    meta = obj.meta
    num = int(meta['partitions_-size'])
    data = []
    for i in range(num):
        if meta[f'partitions_{i}'].islocal:
            data.append(
                torch.utils.data.TensorDataset(
                    resolver.run(obj.member(f'partitions_{i}'))
                )
            )
    return torch.utils.data.ConcatDataset(data)