Doesn't work with concatenated dataset

[gusti-alfarisy]

Using the ImbalancedDatasetSampler in the concatenated dataset using ConcatDataset([datasetA, datasetB, datasetC])

AttributeError: 'ConcatDataset' object has no attribute 'get_labels'

[Chris321sirhc]

I have the same issue, what is the workaround if I need to use concatenated datasets?