Based on the provided stack trace it seems the NotImplementedError is raised by the ImbalancedDatasetSampler, so I would recommend to create an issue in the project’s GitHub repository for better visibility.